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
67 lines
2.9 KiB
MLIR
67 lines
2.9 KiB
MLIR
// RUN: mlir-opt %s -transform-interpreter -canonicalize -split-input-file | FileCheck %s -check-prefix=TILE-10n25
|
|
|
|
func.func @indexed_vector(%arg0: memref<50xindex>) {
|
|
linalg.generic {indexing_maps = [affine_map<(i) -> (i)>],
|
|
iterator_types = ["parallel"]}
|
|
outs(%arg0 : memref<50xindex>) {
|
|
^bb0(%a: index):
|
|
%i = linalg.index 0 : index
|
|
linalg.yield %i : index
|
|
}
|
|
return
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
|
|
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
|
|
%1, %loop = transform.structured.tile_using_for %0 tile_sizes [10] : (!transform.any_op) -> (!transform.any_op, !transform.any_op)
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// TILE-10n25-DAG: [[$MAP:#[a-zA-Z0-9_]*]] = affine_map<(d0)[s0] -> (d0 + s0)>
|
|
// TILE-10n25-LABEL: func @indexed_vector
|
|
// TILE-10n25: %[[C10:.*]] = arith.constant 10 : index
|
|
// TILE-10n25: scf.for %[[J:.*]] = {{.*}} step %[[C10]]
|
|
// TILE-10n25: linalg.generic
|
|
// TILE-10n25: %[[I:.*]] = linalg.index 0 : index
|
|
// TILE-10n25: %[[NEW_I:.*]] = affine.apply [[$MAP]](%[[J]])[%[[I]]]
|
|
// TILE-10n25: linalg.yield %[[NEW_I]] : index
|
|
|
|
// -----
|
|
|
|
func.func @indexed_matrix(%arg0: memref<50x50xindex>) {
|
|
linalg.generic {indexing_maps = [affine_map<(i, j) -> (i, j)>],
|
|
iterator_types = ["parallel", "parallel"]}
|
|
outs(%arg0 : memref<50x50xindex>) {
|
|
^bb0(%a: index):
|
|
%i = linalg.index 0 : index
|
|
%j = linalg.index 1 : index
|
|
%sum = arith.addi %i, %j : index
|
|
linalg.yield %sum : index
|
|
}
|
|
return
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
|
|
%0 = transform.structured.match ops{["linalg.generic"]} in %arg1 : (!transform.any_op) -> !transform.any_op
|
|
%1, %loop:2 = transform.structured.tile_using_for %0 tile_sizes [10, 25] : (!transform.any_op) -> (!transform.any_op, !transform.any_op, !transform.any_op)
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// TILE-10n25-DAG: [[$MAP:#[a-zA-Z0-9_]*]] = affine_map<(d0)[s0] -> (d0 + s0)>
|
|
// TILE-10n25-LABEL: func @indexed_matrix
|
|
// TILE-10n25-DAG: %[[C25:.*]] = arith.constant 25 : index
|
|
// TILE-10n25-DAG: %[[C10:.*]] = arith.constant 10 : index
|
|
// TILE-10n25: scf.for %[[K:.*]] = {{.*}} step %[[C10]]
|
|
// TILE-10n25: scf.for %[[L:.*]] = {{.*}} step %[[C25]]
|
|
// TILE-10n25: linalg.generic
|
|
// TILE-10n25: %[[I:.*]] = linalg.index 0 : index
|
|
// TILE-10n25: %[[NEW_I:.*]] = affine.apply [[$MAP]](%[[K]])[%[[I]]]
|
|
// TILE-10n25: %[[J:.*]] = linalg.index 1 : index
|
|
// TILE-10n25: %[[NEW_J:.*]] = affine.apply [[$MAP]](%[[L]])[%[[J]]]
|
|
// TILE-10n25: %[[SUM:.*]] = arith.addi %[[NEW_I]], %[[NEW_J]] : index
|
|
// TILE-10n25: linalg.yield %[[SUM]] : index
|