mirror of https://github.com/llvm/torch-mlir
parent
55bbbec144
commit
30017af5b5
1
setup.py
1
setup.py
|
@ -76,6 +76,7 @@ class CMakeBuild(build_py):
|
|||
f"-DMLIR_ENABLE_BINDINGS_PYTHON=ON",
|
||||
f"-DLLVM_ENABLE_PROJECTS=mlir",
|
||||
f"-DTORCH_MLIR_ENABLE_MHLO=ON",
|
||||
f"-DTORCH_MLIR_ENABLE_LTC=OFF",
|
||||
f"-DLLVM_EXTERNAL_PROJECTS=torch-mlir;torch-mlir-dialects",
|
||||
f"-DLLVM_EXTERNAL_TORCH_MLIR_SOURCE_DIR={src_dir}",
|
||||
f"-DLLVM_EXTERNAL_TORCH_MLIR_DIALECTS_SOURCE_DIR={src_dir}/externals/llvm-external-projects/torch-mlir-dialects",
|
||||
|
|
Loading…
Reference in New Issue