[QST] [CuTeDSL] Custom Data Structure: NamedTuple

[HanGuo97]

What is your question?

Is there an easy way to pass a custom NamedTuple into the kernel?

From this function, it seems like tuple/list is supported, but that construction would fail for NamedTuple.

Thanks in advance!

[anakinxc]

Hi @HanGuo97

Thanks for the suggestion, we will add NamedTuple support.

[HanGuo97]

Thank you! I've been looking at how quack handles this. Is this the recommended way?

More specifically:

• Is it necessarily to filter out static vs dynamic entries?
• Do we need to separately define a structure for host struct and kernel struct? The doc wasn't super clear about this.

Thanks again!

[anakinxc]

Yes, that the right way to add custom data structure support.

Is it necessarily to filter out static vs dynamic entries?

Yes, static entries have no ir values so they have to be skipped during extract/new to/from ir values.

Do we need to separately define a structure for host struct and kernel struct? The doc wasn't super clear about this.

Technically no, but for host struct, if you want to use it as a top level jit argument, it needs to implement JitArgument protocol + DynamicExpression protocol

[HanGuo97]

gotcha, thanks again!

[anakinxc]

@HanGuo97 We should probably leave this issue open so we can track add NamedTuple support

[HanGuo97]

ah okay