Conversation
|
cc @Hzfengsy |
tqchen
left a comment
tqchen
left a comment
There was a problem hiding this comment.
overall looks good, please fix lint
|
|
||
| class TensorMapTypeNode : public TypeNode { | ||
| public: | ||
| void VisitAttrs(AttrVisitor* v) { v->Visit("span", &span); } |
There was a problem hiding this comment.
does not have to be now, see #18098 to transition to new reflection, i can do that after erge as well
|
cc @LeiWang1999 |
|
Should we add a version check? |
|
@w1049 can you elaborate on the failure cases? |
The Similar issue: llvm/llvm-project#64529 |
|
i see, seems we should guard the registration of |
This PR introduces Cutensor map support in the runtime module. It enables calling kernels whose arguments are cuTensorMap, these arguments are passed as handle(address) and associated with arg_extra_tags that indicate indicate it is tensor map. The TensorMap is allocated on stack with a runtime API
4 participants
This pull request enables support for CutensorMap in both the IR and runtime layers.