diff --git a/op_builder/builder.py b/op_builder/builder.py index 13853df92..eb78c28cf 100644 --- a/op_builder/builder.py +++ b/op_builder/builder.py @@ -432,7 +432,7 @@ class OpBuilder(ABC): except MissingCUDAException: print(f"{WARNING} {self.name} cuda is missing or is incompatible with installed torch, " "only cpu ops can be compiled!") - return '-D__DISABLE_CUDA__' + return '-D__DISABLE_CUDA__' def _backup_cpuinfo(self): # Construct cpu_info dict from lscpu that is similar to what py-cpuinfo provides