- 博客(3)
- 收藏
- 关注
原创 昇腾npu上怎么查找UB大小?
我在昇腾社区文档上查到GetCoreMemSize接口可以获取到UB内存空间的大小,但是发现要使用这个接口,必须得先创建个算子工程,在算子工程里面调用这个接口再打印才知道UB的size到底是多大,这样有点太麻烦了。Ascend C的Vector算子输入输出都存在UB这块内存上,写算子的时候就需要知道这块空间到底有多大,够不够用了。
2025-06-20 17:38:22
275
转载 Ascend C算子怎么实现广播算法(1)?
https://blog.csdn.net/weixin_52153243?type=blog
2025-05-26 20:52:42
54
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人