mlir/test/Conversion/AffineToStandard/lower-affine-gpu.mlir

   1 // RUN: mlir-opt -pass-pipeline="builtin.module(gpu.module(lower-affine))" %s | FileCheck %s
   2
   3 #map0gpufunc = affine_map<(d0) -> (d0)>
   4 gpu.module @kernels {
   5   gpu.func @foo(%arg0 : index, %arg1 : memref<?xf32>) -> f32 {
   6     %0 = affine.apply #map0gpufunc(%arg0)
   7     %1 = memref.load %arg1[%0] : memref<?xf32>
   8     gpu.return %1 : f32
   9   }
  10
  11 //      CHECK: gpu.func
  12 // CHECK-SAME: %[[ARG0:.*]]: index
  13 //  CHECK-NOT:   affine.apply
  14 //      CHECK:   memref.load %{{.*}}[%[[ARG0]]]
  15 }