From 754f1549a39181e6d71079c85ecbda614e76e57d Mon Sep 17 00:00:00 2001
From: Maksim Levental <maksim.levental@gmail.com>
Date: Fri, 21 Nov 2025 04:21:35 -0800
Subject: [PATCH] [wheel] set MLIR_BINDINGS_PYTHON_NB_DOMAIN correctly

See https://github.com/llvm/lighthouse/issues/14
---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index afb67496136f..6976e79b539d 100644
--- a/setup.py
+++ b/setup.py
@@ -116,6 +116,7 @@ def cmake_build(self, cmake_build_dir):
             f"-DPython_EXECUTABLE={sys.executable}",
             f"-DPython_FIND_VIRTUALENV=ONLY",
             f"-DMLIR_ENABLE_BINDINGS_PYTHON=ON",
+            "-DMLIR_BINDINGS_PYTHON_NB_DOMAIN=torch_mlir",
             f"-DLLVM_TARGETS_TO_BUILD=host",
             f"-DLLVM_ENABLE_ZSTD=OFF",
             # Optimization options for building wheels.