unit tests for fusion

octoml · Apr 18, 2023 · 752366f · 752366f
1 parent e8e6b0e
commit 752366f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/python/octo/test_octo_flow.py b/tests/python/octo/test_octo_flow.py
@@ -96,6 +96,7 @@ def test_construct_schedule_map():
 
     # Apply the cutlass partitioning pass.
     mod = partition_for_cutlass(relax_mod)
+    print("post PARTITION", mod)
 
     # Construct the schedule map.
     schedule_map = tvm.octo.construct_schedule_map(model, mod)