diff --git a/iree/compiler/Conversion/LinalgToSPIRV/test/BUILD b/iree/compiler/Conversion/LinalgToSPIRV/test/BUILD index 8ed1c59a3487..761d17431f4a 100644 --- a/iree/compiler/Conversion/LinalgToSPIRV/test/BUILD +++ b/iree/compiler/Conversion/LinalgToSPIRV/test/BUILD @@ -28,6 +28,7 @@ iree_lit_test_suite( srcs = enforce_glob( [ "batch_matmul_vectorization.mlir", + "concretize_tile_among_workgroups.mlir", "convert_to_gpu.mlir", "convert_to_spirv.mlir", "dead_alloc.mlir", @@ -43,6 +44,8 @@ iree_lit_test_suite( "pipeline_test.mlir", "pipeline_test_cooperative_mat.mlir", "split_dispatch_function.mlir", + "tile_and_vectorize_conv.mlir", + "tile_and_vectorize_matmul.mlir", "vector_to_gpu.mlir", "workgroup_memory_promotion.mlir", ], diff --git a/iree/compiler/Conversion/LinalgToSPIRV/test/CMakeLists.txt b/iree/compiler/Conversion/LinalgToSPIRV/test/CMakeLists.txt index 30df4bd36c9e..9c7c5001e05f 100644 --- a/iree/compiler/Conversion/LinalgToSPIRV/test/CMakeLists.txt +++ b/iree/compiler/Conversion/LinalgToSPIRV/test/CMakeLists.txt @@ -15,6 +15,7 @@ iree_lit_test_suite( lit SRCS "batch_matmul_vectorization.mlir" + "concretize_tile_among_workgroups.mlir" "convert_to_gpu.mlir" "convert_to_spirv.mlir" "dead_alloc.mlir" @@ -30,6 +31,8 @@ iree_lit_test_suite( "pipeline_test.mlir" "pipeline_test_cooperative_mat.mlir" "split_dispatch_function.mlir" + "tile_and_vectorize_conv.mlir" + "tile_and_vectorize_matmul.mlir" "vector_to_gpu.mlir" "workgroup_memory_promotion.mlir" DATA