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
65 lines
3.3 KiB
MLIR
65 lines
3.3 KiB
MLIR
// Note: We run CSE here to make the pattern matching more direct.
|
|
|
|
// RUN: mlir-opt %s -test-lower-to-llvm -cse -canonicalize | FileCheck %s
|
|
|
|
// RUN: mlir-opt %s \
|
|
// RUN: -transform-preload-library=transform-library-paths=%p/../Transform/include/Library/lower-to-llvm.mlir \
|
|
// RUN: -transform-interpreter="entry-point=entry_point" \
|
|
// RUN: -test-transform-dialect-erase-schedule \
|
|
// RUN: -cse -canonicalize \
|
|
// RUN: | FileCheck %s
|
|
|
|
// Check that we properly lower to llvm memref operations that require to be
|
|
// expanded first, like `memref.subview`.
|
|
func.func @subview(%0 : memref<64x4xf32, strided<[4, 1], offset: 0>>, %arg0 : index, %arg1 : index, %arg2 : index)
|
|
-> memref<?x?xf32, strided<[?, ?], offset: ?>> {
|
|
// CHECK-LABEL: @subview
|
|
// CHECK-SAME: %[[BASE:[^:]*]]: !llvm.ptr
|
|
// CHECK-SAME: %[[BASE_ALIGNED:[^:]*]]: !llvm.ptr,
|
|
// CHECK-SAME: %[[BASE_OFFSET:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[BASE_STRIDE0:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[BASE_STRIDE1:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[BASE_SIZE0:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[BASE_SIZE1:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[ARG0:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[ARG1:[^:]*]]: i64,
|
|
// CHECK-SAME: %[[ARG2:[^:]*]]: i64)
|
|
// CHECK-SAME: -> !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>
|
|
|
|
// CHECK-DAG: %[[STRIDE0:.*]] = llvm.mlir.constant(4 : index) : i64
|
|
// CHECK-DAG: %[[DESCSTRIDE0:.*]] = llvm.mul %[[ARG0]], %[[STRIDE0]] overflow<nsw> : i64
|
|
// CHECK-DAG: %[[OFF2:.*]] = llvm.add %[[DESCSTRIDE0]], %[[ARG1]] : i64
|
|
// CHECK-DAG: %[[DESC:.*]] = llvm.mlir.poison : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
|
|
// Base address and algined address.
|
|
// CHECK-DAG: %[[DESC0:.*]] = llvm.insertvalue %[[BASE]], %[[DESC]][0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
// CHECK-DAG: %[[DESC1:.*]] = llvm.insertvalue %[[BASE_ALIGNED]], %[[DESC0]][1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
|
|
// Offset.
|
|
// CHECK: %[[DESC2:.*]] = llvm.insertvalue %[[OFF2]], %[[DESC1]][2] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
// Size 0.
|
|
// CHECK: %[[DESC3:.*]] = llvm.insertvalue %[[ARG0]], %[[DESC2]][3, 0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
// Stride 0 == 4 * %arg0.
|
|
// CHECK: %[[DESC4:.*]] = llvm.insertvalue %[[DESCSTRIDE0]], %[[DESC3]][4, 0] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
// Size 1.
|
|
// CHECK: %[[DESC5:.*]] = llvm.insertvalue %[[ARG1]], %[[DESC4]][3, 1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
// Stride 1 == 1 * %arg1.
|
|
// CHECK: %[[DESC6:.*]] = llvm.insertvalue %[[ARG1]], %[[DESC5]][4, 1] : !llvm.struct<(ptr, ptr, i64, array<2 x i64>, array<2 x i64>)>
|
|
|
|
%1 = memref.subview %0[%arg0, %arg1][%arg0, %arg1][%arg0, %arg1] :
|
|
memref<64x4xf32, strided<[4, 1], offset: 0>>
|
|
to memref<?x?xf32, strided<[?, ?], offset: ?>>
|
|
return %1 : memref<?x?xf32, strided<[?, ?], offset: ?>>
|
|
}
|
|
|
|
module @named_inclusion_in_named attributes { transform.with_named_sequence } {
|
|
transform.named_sequence private @lower_to_llvm(!transform.any_op {transform.readonly}) -> !transform.any_op
|
|
|
|
transform.named_sequence @entry_point(
|
|
%toplevel_module : !transform.any_op {transform.readonly}) {
|
|
transform.include @lower_to_llvm failures(suppress) (%toplevel_module)
|
|
: (!transform.any_op) -> (!transform.any_op)
|
|
transform.yield
|
|
}
|
|
}
|