3
数学库
rocWMMA适配情况
2025-08-04 22:20:57
51

目前虽然能在K100_AI上用tensor core进行计算,但是这样直接用汇编编程比较低效易错,我了解到AMD有对标英伟达WMMA的rocWMMA,请问rocWMMA是有有海光适配?如何获取使用?

全部评论(2)
只看楼主
T
tianlh
2楼
后续DTK版本会发布对标的tensorcore模板库
2025/8/5
T
tianlh
1楼
目前没有适配
2025/8/5
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据