diff options
| author | Deval Shah <[email protected]> | 2019-02-14 15:15:29 -0800 |
|---|---|---|
| committer | Deval Shah <[email protected]> | 2019-02-14 15:15:29 -0800 |
| commit | 431381ece357b1bfae7171743e56ab9361b749f4 (patch) | |
| tree | 20216dfed6fcb99160d771524c174f9c6cb028c2 /libcuda | |
| parent | 9fb5819697e4caa107980635887226d507f9cb36 (diff) | |
Skip app version check for pytorch
Diffstat (limited to 'libcuda')
| -rw-r--r-- | libcuda/cuda_runtime_api.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/libcuda/cuda_runtime_api.cc b/libcuda/cuda_runtime_api.cc index 0f7aac2..e6b53c3 100644 --- a/libcuda/cuda_runtime_api.cc +++ b/libcuda/cuda_runtime_api.cc @@ -2583,10 +2583,10 @@ void** CUDARTAPI __cudaRegisterFatBinary( void *fatCubin ) // with a newer version of CUDA to run apps compiled with older versions of // CUDA. This is especially useful for PTXPLUS execution. //Skip cuda version check for pytorch application - std::string app_binary_path = "ldd " + get_app_binary(); + std::string app_binary_path = get_app_binary(); int pos = app_binary_path.find("python"); if (pos==std::string::npos){ - printf("Not pytorch app : checking cuda version"); + // Not pytorch app : checking cuda version int app_cuda_version = get_app_cuda_version(); assert( app_cuda_version == CUDART_VERSION / 1000 && "The app must be compiled with same major version as the simulator." ); } |
