xref: /llvm-project/mlir/test/Conversion/AffineToStandard/lower-affine-gpu.mlir (revision 13bd41096286305ee603428f6adf161f52981827) 
1// RUN: mlir-opt -pass-pipeline="builtin.module(gpu.module(lower-affine))" %s | FileCheck %s
2
3#map0gpufunc = affine_map<(d0) -> (d0)>
4gpu.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}
16