我正在尝试在我的本地机器上使用OpenMP将工作传送到我的GTX 1060显卡上。我的所有CUDA和Cublas示例都可以正常运行。但是,当我尝试运行一些OpenMP卸载时,它根本不起作用。为了支持OpenMP 5.0,我编译了GCC 10.2.0工具链。经过一些调试,我发现OpenMP运行时没有看到任何设备。例如,此代码显示为零:
#include <omp.h>
#include <stdio.h>
int main() {
printf("%d\n", omp_get_num_devices());
return 0;
}
不过,Nvidia工具链已经可用并运转正常:
$ nvidia-smi
Sun Feb 21 23:06:40 2021
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.51.06 Driver Version: 450.51.06 CUDA Version: 11.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce GTX 106... Off | 00000000:1D:00.0 Off | N/A |
| 0% 37C P8 12W / 200W | 584MiB / 6075MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
那我错过了什么?OpenMP运行时如何找到设备?
编辑:我在附加关于我的编译器的信息。
$ gcc -v
Using built-in specs.
COLLECT_GCC=gcc
COLLECT_LTO_WRAPPER=/opt/gcc/10.2.0/libexec/gcc/x86_64-pc-linux-gnu/10.2.0/lto-wrapper
Target: x86_64-pc-linux-gnu
Configured with: ./configure --prefix=/opt/gcc/10.2.0/
Thread model: posix
Supported LTO compression algorithms: zlib
gcc version 10.2.0 (GCC)
使用以下命令编译代码:
gcc -fopenmp simple.c
-fopenmp
无法启用卸载功能。您还需要传递-foffload=nvptx-none
,告诉GCC您想要将卸载任务分配到NVIDIA设备上。如果出现nvptx-none
不支持的错误,则说明您的GCC没有构建对它的支持。gcc -v
显示构建配置信息。 - Hristo Iliev