Numba支持在AMD设备上声明本地内存吗?

问题描述

我有一些最初编写为与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 (将#修改为@)