File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed
Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -462,11 +462,12 @@ def jit_load(self, verbose=True):
462462 raise RuntimeError (f"Unable to JIT load the { self .name } op due to ninja not being installed." )
463463
464464 if isinstance (self , CUDAOpBuilder ) and not self .is_rocm_pytorch ():
465- try :
466- assert_no_cuda_mismatch (self .name )
467- self .build_for_cpu = False
468- except BaseException :
469- self .build_for_cpu = True
465+ #TODO(jeff): need to come back and fix cpu-only builds, this came in on #3085 but is hiding real user env issues (eg. torch cuda != sys cuda)
466+ #try:
467+ assert_no_cuda_mismatch (self .name )
468+ self .build_for_cpu = False
469+ #except BaseException:
470+ # self.build_for_cpu = True
470471
471472 self .jit_mode = True
472473 from torch .utils .cpp_extension import load
You can’t perform that action at this time.
0 commit comments