Enable torch_mlir_e2e_test when TORCH_MLIR_ENABLE_JIT_IR_IMPORTER OFF

- To fix the bug No module named 'torch_mlir_e2e_test' when test torch-stable
llvm · Aug 21, 2024 · 3cdae9f · 3cdae9f
1 parent 02ed307
commit 3cdae9f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/projects/pt1/python/CMakeLists.txt b/projects/pt1/python/CMakeLists.txt
@@ -41,9 +41,11 @@ endif()
 if(TORCH_MLIR_ENABLE_JIT_IR_IMPORTER)
   add_subdirectory(torch_mlir/jit_ir_importer)
   add_subdirectory(torch_mlir/csrc/jit_ir_importer)
+else()
   add_subdirectory(torch_mlir_e2e_test)
 endif()
 
+
 ################################################################################
 # Custom op example
 # Required for running the update_torch_ods.sh and update_abstract_interp_lib.sh
@@ -61,6 +63,10 @@ if(TORCH_MLIR_ENABLE_JIT_IR_IMPORTER)
     TorchMLIRJITIRImporterPybind
     TorchMLIRE2ETestPythonModules
   )
+else()
+  add_dependencies(TorchMLIRPythonTorchExtensionsSources
+    TorchMLIRE2ETestPythonModules
+  )
 endif()
 
 if(TORCH_MLIR_ENABLE_LTC)