// RUN: mlir-opt %s -convert-gpu-to-llvm-spv | FileCheck %s

gpu.module @kernels {
  // CHECK-LABEL: llvm.func spir_kernelcc @constant_load
  // Constant address space maps to SPIRV/OpenCL address space 2 (UniformConstant)
  // CHECK-SAME: !llvm.ptr<2>
  gpu.func @constant_load(%arg0: memref<16xf32, #gpu.address_space<constant>>) kernel {
    %c0 = arith.constant 0 : index
    %v = memref.load %arg0[%c0] : memref<16xf32, #gpu.address_space<constant>>
    gpu.return
  }

  // CHECK-LABEL: llvm.func spir_funccc @all_address_spaces
  // Global -> 1, Workgroup -> 3, Private -> 0 (default), Constant -> 2
  // CHECK-SAME: !llvm.ptr<1>
  // CHECK-SAME: !llvm.ptr<3>
  // CHECK-SAME: !llvm.ptr,
  // CHECK-SAME: !llvm.ptr<2>
  gpu.func @all_address_spaces(
    %arg0: memref<f32, #gpu.address_space<global>>,
    %arg1: memref<f32, #gpu.address_space<workgroup>>,
    %arg2: memref<f32, #gpu.address_space<private>>,
    %arg3: memref<f32, #gpu.address_space<constant>>) {
    gpu.return
  }
}