问题描述
我有一些最初编写为与CUDA一起运行的代码,并利用声明的本地内存,如下所示:
numba.cuda.local.array(shape,type)
太好了。
我现在正在将此代码移植到ROCm上,在这里Numba的文档中没有看到类似的内容:
https://numba.pydata.org/numba-doc/latest/roc/memory.html
但是后来我注意到函数numba.rocm.barrier
的文档说:“ ...分别访问全局内存或本地内存时。” -引用本地内存。
所以我的问题是:有没有办法在Rocm上用Numba声明类似于cuda.local.array
的本地数组?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)