summaryrefslogtreecommitdiff
path: root/libcuda
diff options
context:
space:
mode:
authorDeval Shah <[email protected]>2019-02-14 15:15:29 -0800
committerDeval Shah <[email protected]>2019-02-14 15:15:29 -0800
commit431381ece357b1bfae7171743e56ab9361b749f4 (patch)
tree20216dfed6fcb99160d771524c174f9c6cb028c2 /libcuda
parent9fb5819697e4caa107980635887226d507f9cb36 (diff)
Skip app version check for pytorch
Diffstat (limited to 'libcuda')
-rw-r--r--libcuda/cuda_runtime_api.cc4
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." );
}