summaryrefslogtreecommitdiff
path: root/mlir/test/python/execution_engine.py
diff options
context:
space:
mode:
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