File tree Expand file tree Collapse file tree 1 file changed +18
-0
lines changed
test/mlir/test/gc/Transforms/GPU Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Original file line number Diff line number Diff line change 1+ // RUN: gc-opt %s --gc-gpu-pipeline -split-input-file | FileCheck %s
2+
3+ // Regression check for infinite loop in the linalg-to-xegpu pass
4+ // CHECK-DAG: @gcGpuOclKernel_corner_shape_matmul_f16_kernel_SPIRV
5+ // CHECK-DAG: @createGcGpuOclKernel_corner_shape_matmul_f16_kernel
6+ // CHECK-DAG: @corner_shape_matmul_f16
7+ module @fragment_name attributes {} {
8+ func.func @corner_shape_matmul_f16 (%arg0: memref <521 x521 xf16 >, %arg1: memref <521 x521 xf16 >, %arg2: memref <521 x521 xf16 >) {
9+ %0 = bufferization.to_tensor %arg0 restrict : memref <521 x521 xf16 >
10+ %1 = bufferization.to_tensor %arg1 restrict : memref <521 x521 xf16 >
11+ %2 = tensor.empty () : tensor <521 x521 xf16 >
12+ %cst = arith.constant 0.000000e+00 : f16
13+ %3 = linalg.fill ins (%cst : f16 ) outs (%2 : tensor <521 x521 xf16 >) -> tensor <521 x521 xf16 >
14+ %4 = linalg.matmul_transpose_b ins (%0 , %1 : tensor <521 x521 xf16 >, tensor <521 x521 xf16 >) outs (%3 : tensor <521 x521 xf16 >) -> tensor <521 x521 xf16 >
15+ bufferization.materialize_in_destination %4 in restrict writable %arg2 : (tensor <521 x521 xf16 >, memref <521 x521 xf16 >) -> ()
16+ return
17+ }
18+ }
You can’t perform that action at this time.
0 commit comments