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
78 lines
3.1 KiB
MLIR
78 lines
3.1 KiB
MLIR
// RUN: mlir-opt %s | mlir-opt | FileCheck %s
|
|
|
|
// CHECK-LABEL: @opaque_ptr_load
|
|
llvm.func @opaque_ptr_load(%arg0: !llvm.ptr) -> i32 {
|
|
// CHECK: = llvm.load %{{.*}} : !llvm.ptr -> i32
|
|
%0 = llvm.load %arg0 : !llvm.ptr -> i32
|
|
llvm.return %0 : i32
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_store
|
|
llvm.func @opaque_ptr_store(%arg0: i32, %arg1: !llvm.ptr){
|
|
// CHECK: llvm.store %{{.*}}, %{{.*}} : i32, !llvm.ptr
|
|
llvm.store %arg0, %arg1 : i32, !llvm.ptr
|
|
llvm.return
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_ptr_store
|
|
llvm.func @opaque_ptr_ptr_store(%arg0: !llvm.ptr, %arg1: !llvm.ptr) {
|
|
// CHECK: llvm.store %{{.*}}, %{{.*}} : !llvm.ptr, !llvm.ptr
|
|
llvm.store %arg0, %arg1 : !llvm.ptr, !llvm.ptr
|
|
llvm.return
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_alloca
|
|
llvm.func @opaque_ptr_alloca(%arg0: i32) -> !llvm.ptr {
|
|
// CHECK: llvm.alloca %{{.*}} x f32 : (i32) -> !llvm.ptr
|
|
%0 = llvm.alloca %arg0 x f32 : (i32) -> !llvm.ptr
|
|
llvm.return %0 : !llvm.ptr
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_gep
|
|
llvm.func @opaque_ptr_gep(%arg0: !llvm.ptr, %arg1: i32) -> !llvm.ptr {
|
|
// CHECK: = llvm.getelementptr %{{.*}}[%{{.*}}] : (!llvm.ptr, i32) -> !llvm.ptr, f32
|
|
%0 = llvm.getelementptr %arg0[%arg1] : (!llvm.ptr, i32) -> !llvm.ptr, f32
|
|
llvm.return %0 : !llvm.ptr
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_gep_struct
|
|
llvm.func @opaque_ptr_gep_struct(%arg0: !llvm.ptr, %arg1: i32) -> !llvm.ptr {
|
|
// CHECK: = llvm.getelementptr %{{.*}}[%{{.*}}, 0, 1]
|
|
// CHECK: : (!llvm.ptr, i32) -> !llvm.ptr, !llvm.struct<(struct<(f32, f64)>, struct<(i32, i64)>)>
|
|
%0 = llvm.getelementptr %arg0[%arg1, 0, 1] : (!llvm.ptr, i32) -> !llvm.ptr, !llvm.struct<(struct<(f32, f64)>, struct<(i32, i64)>)>
|
|
llvm.return %0 : !llvm.ptr
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_matrix_load_store
|
|
llvm.func @opaque_ptr_matrix_load_store(%ptr: !llvm.ptr, %stride: i64) -> vector<48 x f32> {
|
|
// CHECK: = llvm.intr.matrix.column.major.load
|
|
// CHECK: vector<48xf32> from !llvm.ptr stride i64
|
|
%0 = llvm.intr.matrix.column.major.load %ptr, <stride=%stride>
|
|
{ isVolatile = 0: i1, rows = 3: i32, columns = 16: i32} :
|
|
vector<48 x f32> from !llvm.ptr stride i64
|
|
// CHECK: llvm.intr.matrix.column.major.store
|
|
// CHECK: vector<48xf32> to !llvm.ptr stride i64
|
|
llvm.intr.matrix.column.major.store %0, %ptr, <stride=%stride>
|
|
{ isVolatile = 0: i1, rows = 3: i32, columns = 16: i32} :
|
|
vector<48 x f32> to !llvm.ptr stride i64
|
|
llvm.return %0 : vector<48 x f32>
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_masked_load
|
|
llvm.func @opaque_ptr_masked_load(%arg0: !llvm.ptr, %arg1: vector<7xi1>) -> vector<7xf32> {
|
|
// CHECK: = llvm.intr.masked.load
|
|
// CHECK: (!llvm.ptr, vector<7xi1>) -> vector<7xf32>
|
|
%0 = llvm.intr.masked.load %arg0, %arg1 { alignment = 1: i32} :
|
|
(!llvm.ptr, vector<7xi1>) -> vector<7xf32>
|
|
llvm.return %0 : vector<7 x f32>
|
|
}
|
|
|
|
// CHECK-LABEL: @opaque_ptr_gather
|
|
llvm.func @opaque_ptr_gather(%M: vector<7x!llvm.ptr>, %mask: vector<7xi1>) -> vector<7xf32> {
|
|
// CHECK: = llvm.intr.masked.gather
|
|
// CHECK: (vector<7x!llvm.ptr>, vector<7xi1>) -> vector<7xf32>
|
|
%a = llvm.intr.masked.gather %M, %mask { alignment = 1: i32} :
|
|
(vector<7x!llvm.ptr>, vector<7xi1>) -> vector<7xf32>
|
|
llvm.return %a : vector<7xf32>
|
|
}
|