Skip to content

Commit 390bfb0

Browse files
Enable batch_gemm test. (#846)
1 parent 005fa11 commit 390bfb0

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

test/Integration/Dialect/XeTile/batch_gemm.mlir

+3-6
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,12 @@
1-
// TODO: replace "--no-mlir-runner" with "--runner imex-cpu-runner" once xesim simulator is available in CI.
21
// RUN: %python_executable %imex_runner --requires=l0-runtime -i %s --pass-pipeline-file=%p/xetile-to-func-vc.pp \
3-
// RUN: --no-mlir-runner -e main \
2+
// RUN: --runner imex-cpu-runner -e main \
43
// RUN: --entry-point-result=void \
54
// RUN: --shared-libs=%irunner_utils,%mlir_runner_utils,%mlir_c_runner_utils,%levelzero_runtime --filecheck
65
// RUN: %python_executable %imex_runner --requires=sycl-runtime -i %s --pass-pipeline-file=%p/xetile-to-func-vc.pp \
7-
// RUN: --no-mlir-runner -e main \
6+
// RUN: --runner imex-cpu-runner -e main \
87
// RUN: --entry-point-result=void \
98
// RUN: --shared-libs=%irunner_utils,%mlir_runner_utils,%mlir_c_runner_utils,%sycl_runtime --filecheck
109

11-
// TODO: Remove this check and enable the check "[ALLCLOSE: TRUE]".
12-
// CHECK: llvm.mlir.global internal constant @b2x3_m128_n256_k96_spirv_binary
1310
module @gemm attributes {gpu.container_module} {
1411
func.func @test(%A: memref<2x3x128x96xf16>, %B: memref<2x3x256x96xf16>) -> memref<2x3x128x256xf32> attributes {llvm.emit_c_interface} {
1512
%c1 = arith.constant 1 : index
@@ -198,7 +195,7 @@ module @gemm attributes {gpu.container_module} {
198195
%cast_C = memref.cast %C : memref<2x3x128x256xf32> to memref<*xf32>
199196
%cast_C_ref = memref.cast %C_ref : memref<2x3x128x256xf32> to memref<*xf32>
200197
// call @printMemrefF32(%cast_C) : (memref<*xf32>) -> ()
201-
// DISABLED-CHECK: [ALLCLOSE: TRUE]
198+
// CHECK: [ALLCLOSE: TRUE]
202199
call @printAllcloseF32(%cast_C, %cast_C_ref) : (memref<*xf32>, memref<*xf32>) -> ()
203200
memref.dealloc %A : memref<2x3x128x96xf16>
204201
memref.dealloc %B : memref<2x3x256x96xf16>

0 commit comments

Comments
 (0)