数学库
是否有load_matrix的海光适配(ds_read?)
2025-09-03 19:50:56
65

为了使用K100_AI的tensor core需要特殊的数据寄存器排布,我知道有__builtin_amdgcn_mmac_f32_16x16x8f32这样的函数来提供对tensor core的方便调用(虽然个人感觉仍然十分不方便)
但是不知道有没有将数据存到寄存器中的函数,我知道有__builtin_amdgcn_ds_read_m32x8f32这个函数,但是并不清楚它的参数和调用方式,不知道有没有相关资料
另外,海光可否尽快推出类似wmma一类的针对tensor core的抽象库来方便从业者进行开发。

全部评论(2)
只看楼主
A
ANYA
1楼
您好,相关内容我们正在添加,后续整理好之后会release出来,也会在帖子上给您同步
2025/9/4
网
网友张顺飞
回复 @ANYA
请问可以先讲解一下__builtin_amdgcn_ds_read_m32x8f32一类ds_read函数的使用说明吗?
2025/9/4
没有更多内容了
--
楼主
Level.4
0 帖子
楼主热帖
暂无数据