minor fix

csarofeen · Sep 21, 2022 · e9c0519 · e9c0519
1 parent 4be509b
commit e9c0519
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch/csrc/jit/codegen/cuda/executor.cpp b/torch/csrc/jit/codegen/cuda/executor.cpp
@@ -113,7 +113,8 @@ std::string FusionExecutor::getStructuredCode(const std::string& kernel) {
 #endif
 #endif
   code += std::string("namespace ") + FusionExecutor::kernelNamespace() +
-      " {\n" + defineNvFuserZero(fusion_->isNvFuserZeroEnabled()) +
+      " {\n" +
+      defineNvFuserZero(fusion_ == nullptr || fusion_->isNvFuserZeroEnabled()) +
       defineIntegerTypes() + defineIndexMode(options_.index_mode) +
       defineComplexTypes() + executor_utils::kernelPreamble() + kernel + "}\n";