From 4f7dd7214b1d81dbbdff826015a67accc10390d2 Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Fri, 3 Nov 2023 20:47:51 -0700 Subject: [PATCH] allow cuda mismatch exceptions to be triggered --- op_builder/builder.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/op_builder/builder.py b/op_builder/builder.py index cd2f71ea560c..cf2d3311667e 100644 --- a/op_builder/builder.py +++ b/op_builder/builder.py @@ -462,11 +462,12 @@ def jit_load(self, verbose=True): raise RuntimeError(f"Unable to JIT load the {self.name} op due to ninja not being installed.") if isinstance(self, CUDAOpBuilder) and not self.is_rocm_pytorch(): - try: - assert_no_cuda_mismatch(self.name) - self.build_for_cpu = False - except BaseException: - self.build_for_cpu = True + #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) + #try: + assert_no_cuda_mismatch(self.name) + self.build_for_cpu = False + #except BaseException: + # self.build_for_cpu = True self.jit_mode = True from torch.utils.cpp_extension import load