16 lines · plain
1// RUN: mlir-opt -pass-pipeline="builtin.module(gpu.module(lower-affine))" %s | FileCheck %s2 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 : f329 }10 11// CHECK: gpu.func12// CHECK-SAME: %[[ARG0:.*]]: index13// CHECK-NOT: affine.apply14// CHECK: memref.load %{{.*}}[%[[ARG0]]]15}16