mlir/test/Conversion/GPUCommon/memref-arg-attrs.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}