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):
462
462
raise RuntimeError (f"Unable to JIT load the { self .name } op due to ninja not being installed." )
463
463
464
464
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
470
471
471
472
self .jit_mode = True
472
473
from torch .utils .cpp_extension import load
You can’t perform that action at this time.
0 commit comments