-
各位大佬们好,看到最新一课中用到的MyModule(实现了向量加法),感觉tir这个粒度已经到了for这种,是不已无法实现向算子库的映射(如cudnn)?想问一下,TVM现在支持cudnn是在哪个层面实现的呢(relay层吗)? |
Beta Was this translation helpful? Give feedback.
Answered by
Hzfengsy
Jun 30, 2022
Replies: 1 comment 1 reply
-
TVM支持cudnn是在图层(relay/relax),可以通过BYOC来实现 |
Beta Was this translation helpful? Give feedback.
1 reply
Answer selected by
eAzure
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
TVM支持cudnn是在图层(relay/relax),可以通过BYOC来实现