File tree Expand file tree Collapse file tree 2 files changed +6
-0
lines changed
Expand file tree Collapse file tree 2 files changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -661,6 +661,11 @@ def check_cuda_runtime():
661661 driver_version = driver_version .value
662662 runtime_version = runtime_version .value
663663
664+ if driver_version == 0 :
665+ # cudart present but no driver detected. Likely isolation
666+ # run such as version check or within a docker build.
667+ return
668+
664669 driver_v = parse (str (driver_version / 1000 ))
665670 runtime_v = parse (str (runtime_version / 1000 ))
666671 # First check the "major" version, known to be incompatible
Original file line number Diff line number Diff line change @@ -1001,6 +1001,7 @@ def apply(self, **kwargs):
10011001 self ._emit_args_profiling ('arguments-preprocess' )
10021002
10031003 # Invoke kernel function with args
1004+ print (args )
10041005 arg_values = [args [p .name ] for p in self .parameters ]
10051006 try :
10061007 with self ._profiler .timer_on ('apply' , comm = args .comm ):
You can’t perform that action at this time.
0 commit comments