commit c2951bafc2516a680907371fabdb3a6c409f1035
author Ivan Butygin <ivan.butygin@gmail.com> Sat Aug 16 09:24:18 2025 +0300
committer GitHub <noreply@github.com> Sat Aug 16 09:24:18 2025 +0300
tree af07abc13017c03dbb33a0046bdc17165b5686f8
parent 006d4be6d7601a17ec06ebd4f682c68b74c515e6

Expose loops transforms through python api (#21710)

Signed-off-by: Ivan Butygin <ivan.butygin@gmail.com>

compiler/bindings/python/CMakeLists.txt

diff --git a/compiler/bindings/python/CMakeLists.txt b/compiler/bindings/python/CMakeLists.txt
index 22a7b64..60493d9 100644
--- a/compiler/bindings/python/CMakeLists.txt
+++ b/compiler/bindings/python/CMakeLists.txt
@@ -234,6 +234,7 @@
   MLIRPythonSources.Dialects.transform.interpreter
   MLIRPythonSources.Dialects.vector
   MLIRPythonSources.Dialects.vector_transform
+  MLIRPythonSources.Dialects.loop_transform
 
   # iree-dialects project.
   IREEDialectsPythonSources

compiler/bindings/python/test/ir/dialects_test.py

diff --git a/compiler/bindings/python/test/ir/dialects_test.py b/compiler/bindings/python/test/ir/dialects_test.py
index 6a14997..86ad071 100644
--- a/compiler/bindings/python/test/ir/dialects_test.py
+++ b/compiler/bindings/python/test/ir/dialects_test.py
@@ -46,6 +46,7 @@
 
 # Smoke test for vector transforms
 from iree.compiler.dialects.transform import vector as vt
+from iree.compiler.dialects.transform import loop
 
 # Make sure that our dialects import.
 from iree.compiler.dialects import flow, hal, stream, vm, util, iree_codegen, iree_gpu