代码之家  ›  专栏  ›  技术社区  ›  Antonio Ragagnin

获取OpenACC可用(或全部)设备内存

  •  0
  • Antonio Ragagnin  · 技术社区  · 4 年前

    我有一个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运行?

    0 回复  |  直到 4 年前
        1
  •  1
  •   Mat Colgrove    4 年前

    请参阅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