mlir/test/Conversion/GPUCommon/memref-arg-attrs.mlir

   1 // RUN: mlir-opt %s -split-input-file -convert-gpu-to-rocdl='use-bare-ptr-memref-call-conv=0' | FileCheck %s --check-prefixes=CHECK,ROCDL
   2 // RUN: mlir-opt %s -split-input-file -convert-gpu-to-nvvm='use-bare-ptr-memref-call-conv=0' | FileCheck %s --check-prefixes=CHECK,NVVM
   3
   4 gpu.module @kernel {
   5   gpu.func @test_func_readonly(%arg0 : memref<f32> {llvm.readonly} ) {
   6     gpu.return
   7   }
   8 }
   9
  10 // CHECK-LABEL:  llvm.func @test_func_readonly
  11 // ROCDL-SAME:  !llvm.ptr {llvm.readonly}
  12 //  NVVM-SAME:  !llvm.ptr {llvm.readonly}
  13
  14
  15 // -----
  16
  17 gpu.module @kernel {
  18   gpu.func @test_func_writeonly(%arg0 : memref<f32> {llvm.writeonly} ) {
  19     gpu.return
  20   }
  21 }
  22
  23 // CHECK-LABEL:  llvm.func @test_func_writeonly
  24 // ROCDL-SAME:  !llvm.ptr {llvm.writeonly}
  25 //  NVVM-SAME:  !llvm.ptr {llvm.writeonly}
  26
  27 // -----
  28
  29 gpu.module @kernel {
  30   gpu.func @test_func_readonly_ptr(%arg0 : !llvm.ptr {llvm.readonly} ) {
  31     gpu.return
  32   }
  33 }
  34
  35 // CHECK-LABEL:  llvm.func @test_func_readonly_ptr
  36 // ROCDL-SAME:  !llvm.ptr {llvm.readonly}
  37 //  NVVM-SAME:  !llvm.ptr {llvm.readonly}
  38
  39 // -----
  40
  41 gpu.module @kernel {
  42   gpu.func @test_func_nonnull(%arg0 : memref<f32> {llvm.nonnull} ) {
  43     gpu.return
  44   }
  45 }
  46
  47 // CHECK-LABEL:  llvm.func @test_func_nonnull
  48 // ROCDL-SAME:  !llvm.ptr {llvm.nonnull}
  49 //  NVVM-SAME:  !llvm.ptr {llvm.nonnull}
  50
  51
  52 // -----
  53
  54 gpu.module @kernel {
  55   gpu.func @test_func_dereferenceable(%arg0 : memref<f32> {llvm.dereferenceable = 4 : i64} ) {
  56     gpu.return
  57   }
  58 }
  59
  60 // CHECK-LABEL:  llvm.func @test_func_dereferenceable
  61 // ROCDL-SAME:  !llvm.ptr {llvm.dereferenceable = 4 : i64}
  62 //  NVVM-SAME:  !llvm.ptr {llvm.dereferenceable = 4 : i64}
  63
  64
  65 // -----
  66
  67 gpu.module @kernel {
  68   gpu.func @test_func_dereferenceable_or_null(%arg0 : memref<f32> {llvm.dereferenceable_or_null = 4 : i64} ) {
  69     gpu.return
  70   }
  71 }
  72
  73 // CHECK-LABEL:  llvm.func @test_func_dereferenceable_or_null
  74 // ROCDL-SAME:  !llvm.ptr {llvm.dereferenceable_or_null = 4 : i64}
  75 //  NVVM-SAME:  !llvm.ptr {llvm.dereferenceable_or_null = 4 : i64}
  76
  77 // -----
  78
  79 gpu.module @kernel {
  80   gpu.func @test_func_noundef(%arg0 : memref<f32> {llvm.noundef} ) {
  81     gpu.return
  82   }
  83 }
  84
  85 // CHECK-LABEL:  llvm.func @test_func_noundef
  86 // ROCDL-SAME:  !llvm.ptr {llvm.noundef}
  87 // ROCDL-SAME:  i64 {llvm.noundef}
  88 //  NVVM-SAME:  !llvm.ptr {llvm.noundef}
  89 //  NVVM-SAME:  i64 {llvm.noundef}