undo mlir test pass fixes

KULeuven-MICAS · Dec 17, 2024 · 0e42881 · 0e42881
1 parent c67cf24
commit 0e42881
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 40 deletions.
diff --git a/compiler/tools/snax_opt_main.py b/compiler/tools/snax_opt_main.py
@@ -45,9 +45,6 @@
 from compiler.transforms.test.debug_to_func import DebugToFuncPass
 from compiler.transforms.test.insert_debugs import InsertDebugPass
 from compiler.transforms.test.test_add_mcycle_around_launch import AddMcycleAroundLaunch
-from compiler.transforms.test.test_transform_dialect_erase_schedule import (
-    TestTransformDialectEraseSchedule,
-)
 from compiler.transforms.test_add_mcycle_around_loop import AddMcycleAroundLoopPass
 from compiler.transforms.test_remove_memref_copy import RemoveMemrefCopyPass
 
@@ -127,10 +124,6 @@ def __init__(
         super().register_pass(SnaxBufferize.name, lambda: SnaxBufferize)
         super().register_pass(FuseStreamingRegions.name, lambda: FuseStreamingRegions)
         super().register_pass(AllocToGlobalPass.name, lambda: AllocToGlobalPass)
-        super().register_pass(
-            TestTransformDialectEraseSchedule.name,
-            lambda: TestTransformDialectEraseSchedule,
-        )
 
         # arg handling
         arg_parser = argparse.ArgumentParser(description=description)

diff --git a/compiler/transforms/test/test_transform_dialect_erase_schedule.py b/compiler/transforms/test/test_transform_dialect_erase_schedule.py
diff --git a/kernels/streamer_matmul/Makefile b/kernels/streamer_matmul/Makefile
@@ -9,7 +9,7 @@ TESTS =
 TESTS += quantized_matmul.x
 TESTS += tiled_quantized_matmul.x
 
-SNAXOPTFLAGS = -p test-transform-dialect-erase-schedule,convert-linalg-to-kernel,insert-accfg-op{accelerator=snax_gemmx},dispatch-kernels,convert-linalg-to-stream,fuse-streaming-regions,snax-bufferize,alloc-to-global,set-memory-space,set-memory-layout,realize-memref-casts,insert-sync-barrier,dispatch-regions{nb_cores=2},convert-stream-to-snax-stream,convert-linalg-to-accfg,convert-accfg-to-csr,snax-copy-to-dma,memref-to-snax,snax-to-func,clear-memory-space
+SNAXOPTFLAGS = -p convert-linalg-to-kernel,insert-accfg-op{accelerator=snax_gemmx},dispatch-kernels,convert-linalg-to-stream,fuse-streaming-regions,snax-bufferize,alloc-to-global,set-memory-space,set-memory-layout,realize-memref-casts,insert-sync-barrier,dispatch-regions{nb_cores=2},convert-stream-to-snax-stream,convert-linalg-to-accfg,convert-accfg-to-csr,snax-copy-to-dma,memref-to-snax,snax-to-func,clear-memory-space
 
 CFLAGS += -std=gnu11
 CFLAGS += -Wall -Wextra

diff --git a/runtime/Makefile.rules b/runtime/Makefile.rules
@@ -75,7 +75,7 @@ LDFLAGS += -lsnRuntime
 
 # TRANSFORM Pre-Preprocessing
 
-MLIRTRANSFORM = --pass-pipeline='builtin.module(transform-interpreter{debug-bind-trailing-args=linalg.quantized_matmul})'
+MLIRTRANSFORM = --pass-pipeline='builtin.module(transform-interpreter{debug-bind-trailing-args=linalg.quantized_matmul}, test-transform-dialect-erase-schedule)'
 
 %.mlir: %.transform.mlir
 	$(MLIROPT) $(MLIRTRANSFORM) -o $@ $<
@@ -89,6 +89,7 @@ MLIRPREPROCFLAGS += --mlir-print-local-scope
 MLIRPREPROC2FLAGS = --tosa-to-arith="include-apply-rescale" 
 MLIRPREPROC2FLAGS += --empty-tensor-to-alloc-tensor
 
+MLIRPREPROC3FLAGS += --test-linalg-transform-patterns="test-generalize-pad-tensor"
 MLIRPREPROC3FLAGS += --linalg-generalize-named-ops
 MLIRPREPROC3FLAGS += --empty-tensor-to-alloc-tensor --one-shot-bufferize="bufferize-function-boundaries allow-return-allocs-from-loops function-boundary-type-conversion=identity-layout-map"
 MLIRPREPROC3FLAGS += --mlir-print-op-generic