From 05337fe1a43c4eeda83f22ae0775ccb748373d34 Mon Sep 17 00:00:00 2001 From: makslevental Date: Thu, 28 Aug 2025 11:56:35 -0400 Subject: [PATCH 1/2] [MLIR][Python] fix StandalonePythonModules shipping all dialect modules --- mlir/examples/standalone/test/python/smoketest.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mlir/examples/standalone/test/python/smoketest.py b/mlir/examples/standalone/test/python/smoketest.py index bd40c65d16164..a1c8821882e70 100644 --- a/mlir/examples/standalone/test/python/smoketest.py +++ b/mlir/examples/standalone/test/python/smoketest.py @@ -24,3 +24,10 @@ # CHECK: %[[C:.*]] = arith.constant 2 : i32 # CHECK: standalone.foo %[[C]] : i32 print(str(module)) + +try: + from mlir_standalone.dialects import quant +except ImportError as e: + assert "symbol not found in flat namespace '_mlirTypeIsAAnyQuantizedType'" not in e.args[0] +else: + assert False, "expected exception not raised" \ No newline at end of file From 78396c898fa0ce71c8b744b3877db5932fd45732 Mon Sep 17 00:00:00 2001 From: Maksim Levental Date: Thu, 28 Aug 2025 12:16:38 -0400 Subject: [PATCH 2/2] Update smoketest.py --- mlir/examples/standalone/test/python/smoketest.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mlir/examples/standalone/test/python/smoketest.py b/mlir/examples/standalone/test/python/smoketest.py index a1c8821882e70..96c0f91b2a498 100644 --- a/mlir/examples/standalone/test/python/smoketest.py +++ b/mlir/examples/standalone/test/python/smoketest.py @@ -25,9 +25,11 @@ # CHECK: standalone.foo %[[C]] : i32 print(str(module)) +from mlir_standalone.dialects import quant + try: from mlir_standalone.dialects import quant except ImportError as e: assert "symbol not found in flat namespace '_mlirTypeIsAAnyQuantizedType'" not in e.args[0] else: - assert False, "expected exception not raised" \ No newline at end of file + assert False, "expected exception not raised"