有没有办法在运行应用程序时禁用 PTX 代码到 GPU 程序集的即时编译?
在某些情况下,人们希望在仅 CPU 模式下运行支持 GPU 的应用程序。如果应用程序足够大,并且是该版本的可执行文件的第一次运行,JIT 过程可能需要很长时间(在我公司的 CI/CD 案例中很常见)
CUDA 运行时将遵循环境变量来确定 PTX JIT 是否发生。以下内容应阻止代码中的任何 JIT 活动:
CUDA_DISABLE_PTX_JIT=1 ./my_executable
Run Code Online (Sandbox Code Playgroud)
归档时间: |
|
查看次数: |
540 次 |
最近记录: |