如何解决获取 OpenACC 空闲或总设备内存
我有一个 GPU 代码,它在每次迭代时决定是否可以将迭代卸载到加速器。 OpenACC 成为最好的工具:
void module(struct my_aos *aos,int n_aos){
int criteria = /* check either that n_aos is large enough and that aos[:n_aos] will fit the GPU */
#pragma acc data copy(aos[0:n_aos]) if(criteria)
#pragma acc parallel loop if(criteria)
for(int i = 0; i < n_aos; i++){
/* work on my_aos*/
}
}
我如何提前决定 aos[0:n_aos]
是否适合 GPU 内存?是否有 openacc_get_free_device_memory()
类型的函数?
如果出现内存不足故障,我如何启动设备复制并返回到仅 cpu 运行?
解决方法
请参阅 OpenACC 标准的“3.2.6 acc get property”部分。特别是“acc_property_free_memory”属性。
https://www.openacc.org/sites/default/files/inline-images/Specification/OpenACC-3.1-final.pdf
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。