90 lines
2.2 KiB
MLIR
90 lines
2.2 KiB
MLIR
|
// RUN: mlir-opt %s -split-input-file -convert-gpu-to-rocdl='use-bare-ptr-memref-call-conv=0' | FileCheck %s --check-prefixes=CHECK,ROCDL
|
||
|
|
// RUN: mlir-opt %s -split-input-file -convert-gpu-to-nvvm='use-bare-ptr-memref-call-conv=0' | FileCheck %s --check-prefixes=CHECK,NVVM
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_readonly(%arg0 : memref<f32> {llvm.readonly} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_readonly
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.readonly}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.readonly}
|
||
|
|
|
||
|
|
|
||
|
|
// -----
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_writeonly(%arg0 : memref<f32> {llvm.writeonly} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_writeonly
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.writeonly}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.writeonly}
|
||
|
|
|
||
|
|
// -----
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_readonly_ptr(%arg0 : !llvm.ptr {llvm.readonly} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_readonly_ptr
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.readonly}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.readonly}
|
||
|
|
|
||
|
|
// -----
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_nonnull(%arg0 : memref<f32> {llvm.nonnull} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_nonnull
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.nonnull}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.nonnull}
|
||
|
|
|
||
|
|
|
||
|
|
// -----
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_dereferenceable(%arg0 : memref<f32> {llvm.dereferenceable = 4 : i64} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_dereferenceable
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.dereferenceable = 4 : i64}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.dereferenceable = 4 : i64}
|
||
|
|
|
||
|
|
|
||
|
|
// -----
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_dereferenceable_or_null(%arg0 : memref<f32> {llvm.dereferenceable_or_null = 4 : i64} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_dereferenceable_or_null
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.dereferenceable_or_null = 4 : i64}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.dereferenceable_or_null = 4 : i64}
|
||
|
|
|
||
|
|
// -----
|
||
|
|
|
||
|
|
gpu.module @kernel {
|
||
|
|
gpu.func @test_func_noundef(%arg0 : memref<f32> {llvm.noundef} ) {
|
||
|
|
gpu.return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK-LABEL: llvm.func @test_func_noundef
|
||
|
|
// ROCDL-SAME: !llvm.ptr {llvm.noundef}
|
||
|
|
// ROCDL-SAME: i64 {llvm.noundef}
|
||
|
|
// NVVM-SAME: !llvm.ptr {llvm.noundef}
|
||
|
|
// NVVM-SAME: i64 {llvm.noundef}
|