相当于CUDA的OPENACC函数cudaError_t cudaMemGetInfo

Question

我想知道程序在运行时特定位置的可用内存大小。我想知道OpenACC是否具有与CUDA的cudaMemGetInfo()等效的功能。

Answer 1

OpenACC标准没有此功能，但PGI确实具有您可以使用的OpenACC扩展API调用。 “ acc_get_free_memory”将返回设备上的可用内存量，而“ acc_get_memory”将返回设备上的总内存量。包括“ accel.h”，这是PGI为其OpenACC扩展提供原型的地方。两者都返回一个无符号的long。

虽然我还没有尝试过自己，但是您也可以直接调用“ cudaMemGetInfo”。