Skip to content

Commit 1cf2871

Browse files
authored
[wheel] set MLIR_BINDINGS_PYTHON_NB_DOMAIN correctly (#4385)
See llvm/lighthouse#14
1 parent b834f94 commit 1cf2871

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@ def cmake_build(self, cmake_build_dir):
116116
f"-DPython_EXECUTABLE={sys.executable}",
117117
f"-DPython_FIND_VIRTUALENV=ONLY",
118118
f"-DMLIR_ENABLE_BINDINGS_PYTHON=ON",
119+
"-DMLIR_BINDINGS_PYTHON_NB_DOMAIN=torch_mlir",
119120
f"-DLLVM_TARGETS_TO_BUILD=host",
120121
f"-DLLVM_ENABLE_ZSTD=OFF",
121122
# Optimization options for building wheels.

0 commit comments

Comments
 (0)