Skip to content

Commit 4f7dd72

Browse files
committed
allow cuda mismatch exceptions to be triggered
1 parent 016b796 commit 4f7dd72

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

op_builder/builder.py

+6-5
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)