21 lines · plain
1// RUN: mlir-opt -xegpu-fold-alias-ops -split-input-file %s | FileCheck %s2 3func.func @fold_subview_with_xegpu_create_nd_tdesc(%arg0 : memref<256x256xf32>, %arg1 : index, %arg2 : index, %arg3 : index, %arg4 : index) ->(!xegpu.tensor_desc<8x16xf32>) {4 %subview = memref.subview %arg0[%arg1, %arg2] [32, 32] [1, 1] :5 memref<256x256xf32> to memref<32x32xf32, strided<[256, 1], offset: ?>>6 %0 = xegpu.create_nd_tdesc %subview[%arg3, %arg4] :7 memref<32x32xf32, strided<[256, 1], offset: ?>> -> !xegpu.tensor_desc<8x16xf32>8 return %0 : !xegpu.tensor_desc<8x16xf32>9}10 11// CHECK-DAG: #[[MAP:.+]] = affine_map<()[s0, s1] -> (s0 + s1)>12// CHECK: func @fold_subview_with_xegpu_create_nd_tdesc13// CHECK-SAME: %[[ARG0:[a-zA-Z0-9]+]]: memref<256x256xf32>14// CHECK-SAME: %[[ARG1:[a-zA-Z0-9]+]]: index15// CHECK-SAME: %[[ARG2:[a-zA-Z0-9]+]]: index16// CHECK-SAME: %[[ARG3:[a-zA-Z0-9]+]]: index17// CHECK-SAME: %[[ARG4:[a-zA-Z0-9]+]]: index18// CHECK-DAG: %[[IDX0:.+]] = affine.apply #[[MAP]]()[%[[ARG1]], %[[ARG3]]]19// CHECK-DAG: %[[IDX1:.+]] = affine.apply #[[MAP]]()[%[[ARG2]], %[[ARG4]]]20// CHECK: xegpu.create_nd_tdesc %[[ARG0]][%[[IDX0]], %[[IDX1]]] : memref<256x256xf32> -> !xegpu.tensor_desc<8x16xf32>21