├── CMakeLists.txt ├── CONTRIBUTING.md ├── LICENSE ├── LLVM_VERSION ├── README.md ├── canonicalization_patterns.cc ├── expansion.cc ├── expansion.h ├── loop_nest.cc ├── loop_nest.h ├── mapped_domain.cc ├── mapped_domain.h ├── sair_attributes.cc ├── sair_attributes.h ├── sair_base.td ├── sair_dialect.cc ├── sair_dialect.h ├── sair_op_interfaces.cc ├── sair_op_interfaces.h ├── sair_ops.cc ├── sair_ops.h ├── sair_ops.td ├── sair_opt.cc ├── sair_registration.cc ├── sair_registration.h ├── sair_types.cc ├── sair_types.h ├── sequence.cc ├── sequence.h ├── storage.cc ├── storage.h ├── test ├── CMakeLists.txt ├── canonicalization.mlir ├── default_expansion.mlir ├── default_expansion_invalid.mlir ├── default_loop_nest.mlir ├── default_sequence.mlir ├── default_storage.mlir ├── default_storage_invalid.mlir ├── domain_shape.mlir ├── from_linalg.mlir ├── from_linalg_invalid.mlir ├── inline_trivial_sair_ops.mlir ├── integration │ ├── sair_to_llvm.mlir │ └── sair_to_loop.mlir ├── introduce_loops.mlir ├── introduce_loops_invalid.mlir ├── invalid.mlir ├── lit.cfg.py ├── lit.site.cfg.py.in ├── lower_map_reduce.mlir ├── lower_proj_any.mlir ├── lower_proj_any_invalid.mlir ├── lower_to_llvm.mlir ├── lower_to_map.mlir ├── lower_to_map_invalid.mlir ├── lowering_annotation_roundtrip.mlir ├── mapping_expr.mlir ├── materialize_buffers.mlir ├── materialize_buffers_invalid.mlir ├── materialize_instances.mlir ├── materialize_instances_invalid.mlir ├── normalize_loops.mlir ├── normalize_loops_invalid.mlir ├── passes.cc ├── passes.h ├── passes.td ├── regression │ └── matmul.mlir ├── roundtrip.mlir ├── sequence.mlir └── storage.mlir ├── transforms ├── CMakeLists.txt ├── default_lowering_attributes.cc ├── default_lowering_attributes.h ├── default_lowering_attributes.td ├── domain_utils.cc ├── domain_utils.h ├── inline_trivial_ops.cc ├── introduce_loops.cc ├── lower_map_reduce.cc ├── lower_proj_any.cc ├── lower_to_map.cc ├── lowering.cc ├── lowering.h ├── lowering.td ├── materialize_buffers.cc ├── materialize_instances.cc ├── normalize_loops.cc ├── sair_from_linalg.cc ├── sair_from_linalg.h └── sair_from_linalg.td ├── util.cc └── util.h /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- 1 | Sair currently does *NOT* accept contributions. Stay tuned. 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/LICENSE -------------------------------------------------------------------------------- /LLVM_VERSION: -------------------------------------------------------------------------------- 1 | 8dee997a8558b460b82b23fb43b197d68258baac -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/README.md -------------------------------------------------------------------------------- /canonicalization_patterns.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/canonicalization_patterns.cc -------------------------------------------------------------------------------- /expansion.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/expansion.cc -------------------------------------------------------------------------------- /expansion.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/expansion.h -------------------------------------------------------------------------------- /loop_nest.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/loop_nest.cc -------------------------------------------------------------------------------- /loop_nest.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/loop_nest.h -------------------------------------------------------------------------------- /mapped_domain.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/mapped_domain.cc -------------------------------------------------------------------------------- /mapped_domain.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/mapped_domain.h -------------------------------------------------------------------------------- /sair_attributes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_attributes.cc -------------------------------------------------------------------------------- /sair_attributes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_attributes.h -------------------------------------------------------------------------------- /sair_base.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_base.td -------------------------------------------------------------------------------- /sair_dialect.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_dialect.cc -------------------------------------------------------------------------------- /sair_dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_dialect.h -------------------------------------------------------------------------------- /sair_op_interfaces.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_op_interfaces.cc -------------------------------------------------------------------------------- /sair_op_interfaces.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_op_interfaces.h -------------------------------------------------------------------------------- /sair_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_ops.cc -------------------------------------------------------------------------------- /sair_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_ops.h -------------------------------------------------------------------------------- /sair_ops.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_ops.td -------------------------------------------------------------------------------- /sair_opt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_opt.cc -------------------------------------------------------------------------------- /sair_registration.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_registration.cc -------------------------------------------------------------------------------- /sair_registration.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_registration.h -------------------------------------------------------------------------------- /sair_types.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_types.cc -------------------------------------------------------------------------------- /sair_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sair_types.h -------------------------------------------------------------------------------- /sequence.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sequence.cc -------------------------------------------------------------------------------- /sequence.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/sequence.h -------------------------------------------------------------------------------- /storage.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/storage.cc -------------------------------------------------------------------------------- /storage.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/storage.h -------------------------------------------------------------------------------- /test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/CMakeLists.txt -------------------------------------------------------------------------------- /test/canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/canonicalization.mlir -------------------------------------------------------------------------------- /test/default_expansion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/default_expansion.mlir -------------------------------------------------------------------------------- /test/default_expansion_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/default_expansion_invalid.mlir -------------------------------------------------------------------------------- /test/default_loop_nest.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/default_loop_nest.mlir -------------------------------------------------------------------------------- /test/default_sequence.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/default_sequence.mlir -------------------------------------------------------------------------------- /test/default_storage.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/default_storage.mlir -------------------------------------------------------------------------------- /test/default_storage_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/default_storage_invalid.mlir -------------------------------------------------------------------------------- /test/domain_shape.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/domain_shape.mlir -------------------------------------------------------------------------------- /test/from_linalg.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/from_linalg.mlir -------------------------------------------------------------------------------- /test/from_linalg_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/from_linalg_invalid.mlir -------------------------------------------------------------------------------- /test/inline_trivial_sair_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/inline_trivial_sair_ops.mlir -------------------------------------------------------------------------------- /test/integration/sair_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/integration/sair_to_llvm.mlir -------------------------------------------------------------------------------- /test/integration/sair_to_loop.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/integration/sair_to_loop.mlir -------------------------------------------------------------------------------- /test/introduce_loops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/introduce_loops.mlir -------------------------------------------------------------------------------- /test/introduce_loops_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/introduce_loops_invalid.mlir -------------------------------------------------------------------------------- /test/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/invalid.mlir -------------------------------------------------------------------------------- /test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lit.cfg.py -------------------------------------------------------------------------------- /test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /test/lower_map_reduce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lower_map_reduce.mlir -------------------------------------------------------------------------------- /test/lower_proj_any.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lower_proj_any.mlir -------------------------------------------------------------------------------- /test/lower_proj_any_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lower_proj_any_invalid.mlir -------------------------------------------------------------------------------- /test/lower_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lower_to_llvm.mlir -------------------------------------------------------------------------------- /test/lower_to_map.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lower_to_map.mlir -------------------------------------------------------------------------------- /test/lower_to_map_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lower_to_map_invalid.mlir -------------------------------------------------------------------------------- /test/lowering_annotation_roundtrip.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/lowering_annotation_roundtrip.mlir -------------------------------------------------------------------------------- /test/mapping_expr.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/mapping_expr.mlir -------------------------------------------------------------------------------- /test/materialize_buffers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/materialize_buffers.mlir -------------------------------------------------------------------------------- /test/materialize_buffers_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/materialize_buffers_invalid.mlir -------------------------------------------------------------------------------- /test/materialize_instances.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/materialize_instances.mlir -------------------------------------------------------------------------------- /test/materialize_instances_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/materialize_instances_invalid.mlir -------------------------------------------------------------------------------- /test/normalize_loops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/normalize_loops.mlir -------------------------------------------------------------------------------- /test/normalize_loops_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/normalize_loops_invalid.mlir -------------------------------------------------------------------------------- /test/passes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/passes.cc -------------------------------------------------------------------------------- /test/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/passes.h -------------------------------------------------------------------------------- /test/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/passes.td -------------------------------------------------------------------------------- /test/regression/matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/regression/matmul.mlir -------------------------------------------------------------------------------- /test/roundtrip.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/roundtrip.mlir -------------------------------------------------------------------------------- /test/sequence.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/sequence.mlir -------------------------------------------------------------------------------- /test/storage.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/test/storage.mlir -------------------------------------------------------------------------------- /transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/CMakeLists.txt -------------------------------------------------------------------------------- /transforms/default_lowering_attributes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/default_lowering_attributes.cc -------------------------------------------------------------------------------- /transforms/default_lowering_attributes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/default_lowering_attributes.h -------------------------------------------------------------------------------- /transforms/default_lowering_attributes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/default_lowering_attributes.td -------------------------------------------------------------------------------- /transforms/domain_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/domain_utils.cc -------------------------------------------------------------------------------- /transforms/domain_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/domain_utils.h -------------------------------------------------------------------------------- /transforms/inline_trivial_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/inline_trivial_ops.cc -------------------------------------------------------------------------------- /transforms/introduce_loops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/introduce_loops.cc -------------------------------------------------------------------------------- /transforms/lower_map_reduce.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/lower_map_reduce.cc -------------------------------------------------------------------------------- /transforms/lower_proj_any.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/lower_proj_any.cc -------------------------------------------------------------------------------- /transforms/lower_to_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/lower_to_map.cc -------------------------------------------------------------------------------- /transforms/lowering.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/lowering.cc -------------------------------------------------------------------------------- /transforms/lowering.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/lowering.h -------------------------------------------------------------------------------- /transforms/lowering.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/lowering.td -------------------------------------------------------------------------------- /transforms/materialize_buffers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/materialize_buffers.cc -------------------------------------------------------------------------------- /transforms/materialize_instances.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/materialize_instances.cc -------------------------------------------------------------------------------- /transforms/normalize_loops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/normalize_loops.cc -------------------------------------------------------------------------------- /transforms/sair_from_linalg.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/sair_from_linalg.cc -------------------------------------------------------------------------------- /transforms/sair_from_linalg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/sair_from_linalg.h -------------------------------------------------------------------------------- /transforms/sair_from_linalg.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/transforms/sair_from_linalg.td -------------------------------------------------------------------------------- /util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/util.cc -------------------------------------------------------------------------------- /util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/structured-additive-IR/HEAD/util.h --------------------------------------------------------------------------------