Skip to content

Commit

Permalink
[Codegen] Remove old attention transformations (#18740)
Browse files Browse the repository at this point in the history
These transformations on iree_linalg_ext.attention have been replaced by
transformations on iree_linalg_ext.online_attention. This path of
attention transformations has been deprecated for a long time and it's
time to delete it.
  • Loading branch information
Groverkss authored Oct 10, 2024
1 parent e50eda3 commit dd3f2a3
Show file tree
Hide file tree
Showing 17 changed files with 5 additions and 1,892 deletions.
6 changes: 0 additions & 6 deletions compiler/src/iree/compiler/Codegen/LLVMGPU/test/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,6 @@ iree_lit_test_suite(
"amdgpu_chained_matmul.mlir",
"amdgpu_contraction_distribution.mlir",
"amdgpu_set_anchor_layouts.mlir",
"attention.mlir",
"attention_mfma.mlir",
"conv_pipeline_test_cuda.mlir",
"conv_pipeline_test_rocm.mlir",
"convert_to_nvvm.mlir",
Expand Down Expand Up @@ -82,8 +80,6 @@ iree_lit_test_suite(
# tensor_dialect_*_spec is a an MLIR file that specifies a
# transformation, it needs to be included as data.
exclude = [
"attention_mfma_transform_spec.mlir",
"attention_transform_spec.mlir",
"transform_dialect_codegen_bufferize_spec.mlir",
"transform_dialect_codegen_foreach_to_gpu_spec.mlir",
"transform_dialect_codegen_vector_distribution_spec.mlir",
Expand All @@ -92,8 +88,6 @@ iree_lit_test_suite(
),
cfg = "//compiler:lit.cfg.py",
data = [
"attention_mfma_transform_spec.mlir",
"attention_transform_spec.mlir",
"transform_dialect_codegen_bufferize_spec.mlir",
"transform_dialect_codegen_foreach_to_gpu_spec.mlir",
"transform_dialect_codegen_vector_distribution_spec.mlir",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,6 @@ iree_lit_test_suite(
"amdgpu_chained_matmul.mlir"
"amdgpu_contraction_distribution.mlir"
"amdgpu_set_anchor_layouts.mlir"
"attention.mlir"
"attention_mfma.mlir"
"cast_address_space_function.mlir"
"cast_type_to_fit_mma.mlir"
"config_matvec.mlir"
Expand Down Expand Up @@ -77,8 +75,6 @@ iree_lit_test_suite(
FileCheck
iree-opt
DATA
attention_mfma_transform_spec.mlir
attention_transform_spec.mlir
transform_dialect_codegen_bufferize_spec.mlir
transform_dialect_codegen_foreach_to_gpu_spec.mlir
transform_dialect_codegen_vector_distribution_spec.mlir
Expand Down
175 changes: 0 additions & 175 deletions compiler/src/iree/compiler/Codegen/LLVMGPU/test/attention.mlir

This file was deleted.

Loading

0 comments on commit dd3f2a3

Please sign in to comment.