derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
52 lines
2.8 KiB
MLIR
52 lines
2.8 KiB
MLIR
// RUN: mlir-opt %s -linalg-block-pack-matmul="block-factors=32,16,64 \
|
|
// RUN: lhs-transpose-outer-blocks=false lhs-transpose-inner-blocks=false \
|
|
// RUN: rhs-transpose-outer-blocks=true rhs-transpose-inner-blocks=true" \
|
|
// RUN: -canonicalize | FileCheck %s --check-prefix=MMT4D
|
|
|
|
// RUN: mlir-opt %s -linalg-block-pack-matmul="block-factors=32,16,64 \
|
|
// RUN: lhs-transpose-outer-blocks=false lhs-transpose-inner-blocks=false \
|
|
// RUN: rhs-transpose-outer-blocks=false rhs-transpose-inner-blocks=false" \
|
|
// RUN: -canonicalize | FileCheck %s --check-prefix=MM4D
|
|
|
|
// RUN: mlir-opt %s -linalg-block-pack-matmul="block-factors=32,16,64 \
|
|
// RUN: lhs-transpose-outer-blocks=true lhs-transpose-inner-blocks=true \
|
|
// RUN: rhs-transpose-outer-blocks=false rhs-transpose-inner-blocks=false" \
|
|
// RUN: -canonicalize | FileCheck %s --check-prefix=MTM4D
|
|
|
|
func.func @block_matmul(
|
|
%A: tensor<64x128xf32>, %B: tensor<128x64xf32>, %C: tensor<64x64xf32>) -> tensor<64x64xf32> {
|
|
%0 = linalg.matmul ins(%A, %B : tensor<64x128xf32>, tensor<128x64xf32>)
|
|
outs(%C : tensor<64x64xf32>) -> tensor<64x64xf32>
|
|
return %0 : tensor<64x64xf32>
|
|
}
|
|
|
|
// MMT4D-DAG: #[[$MAP:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d2, d3, d5)>
|
|
// MMT4D-DAG: #[[$MAP1:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d1, d2, d4, d5)>
|
|
// MMT4D-DAG: #[[$MAP2:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d3, d4)>
|
|
// MMT4D-LABEL: func @block_matmul
|
|
// MMT4D-COUNT-3: linalg.pack
|
|
// MMT4D: linalg.generic
|
|
// MMT4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
|
|
// MMT4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "reduction"]
|
|
// MMT4D-COUNT-1: linalg.unpack
|
|
|
|
// MM4D-DAG: #[[$MAP:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d2, d3, d5)>
|
|
// MM4D-DAG: #[[$MAP1:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d2, d1, d5, d4)>
|
|
// MM4D-DAG: #[[$MAP2:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d3, d4)>
|
|
// MM4D-LABEL: func @block_matmul
|
|
// MM4D-COUNT-3: linalg.pack
|
|
// MM4D: linalg.generic
|
|
// MM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
|
|
// MM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "reduction"]
|
|
// MM4D-COUNT-1: linalg.unpack
|
|
|
|
// MTM4D-DAG: #[[$MAP:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d2, d0, d5, d3)>
|
|
// MTM4D-DAG: #[[$MAP1:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d2, d1, d5, d4)>
|
|
// MTM4D-DAG: #[[$MAP2:.+]] = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d3, d4)>
|
|
// MTM4D-LABEL: func @block_matmul
|
|
// MTM4D-COUNT-3: linalg.pack
|
|
// MTM4D: linalg.generic
|
|
// MTM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
|
|
// MTM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "reduction"]
|
|
// MTM4D-COUNT-1: linalg.unpack
|