summaryrefslogtreecommitdiff
path: root/mlir/test/python/execution_engine.py
diff options
context:
space:
mode:
authorMaksim Levental <maksim.levental@gmail.com>2025-09-27 04:06:35 -0400
committerMaksim Levental <maksim.levental@gmail.com>2025-09-27 23:36:34 -0400
commit7179ffae6e552b5ff7d9fe7c4ad3f66e8b8b3f33 (patch)
tree6e3a4ace77493f58817cc92629e342b8e02efce2 /mlir/test/python/execution_engine.py
parente85d9c80c4dbc77b22a98f6d3ace5af36c35aacc (diff)
undo transform extension stuffusers/makslevental/specific-reg
Diffstat (limited to 'mlir/test/python/execution_engine.py')
-rw-r--r--mlir/test/python/execution_engine.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/mlir/test/python/execution_engine.py b/mlir/test/python/execution_engine.py
index d569fcef32bf..f93337c5cbb6 100644
--- a/mlir/test/python/execution_engine.py
+++ b/mlir/test/python/execution_engine.py
@@ -5,6 +5,7 @@ from mlir.ir import *
from mlir.passmanager import *
from mlir.execution_engine import *
from mlir.runtime import *
+from mlir.dialects import func, memref, arith, complex, cf, llvm
try:
from ml_dtypes import bfloat16, float8_e5m2