Search code examples
cudaptx

Disable CUDA PTX-to-binary JIT compilation


Is there a way to disable the Just-In-Time compilation of PTX code to GPU assembly when running an application?

There are certain scenarios where one want to run a GPU-enabled application on CPU-only mode. If the application is big enough, and it is the first run of that version of executable, the JIT process may take a very long time (very common on CI/CD cases from my company)


Solution

  • The CUDA runtime will obey an environment variable to determine whether PTX JIT will take place or not. The following should prevent any JIT activity in your code:

    CUDA_DISABLE_PTX_JIT=1 ./my_executable