Skip to content

Commit 4b3f550

Browse files
authored
Merge pull request #515 from intel/nishant_enableTest
Enable matmul and complexconv2d test on GPU
2 parents d6d7a38 + 4a7b9c4 commit 4b3f550

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

test/PlaidML/OpTest.ComplexConv2D.mlir

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22
// RUN: --runner mlir-cpu-runner -e main \
33
// RUN: --shared-libs=%mlir_runner_utils,%mlir_c_runner_utils \
44
// RUN: --entry-point-result=void | FileCheck %s
5-
// RUN-GPU: %python_executable %imex_runner -i %s --pass-pipeline-file=%p/linalg-to-llvm.pp \
6-
// RUN-GPU: --runner mlir-cpu-runner -e main \
7-
// RUN-GPU: --entry-point-result=void \
8-
// RUN-GPU: --shared-libs=%mlir_runner_utils,%levelzero_runtime | FileCheck %s
5+
// RUN: %gpu_skip || %python_executable %imex_runner -i %s --pass-pipeline-file=%p/linalg-to-llvm.pp \
6+
// RUN: --runner mlir-cpu-runner -e main \
7+
// RUN: --entry-point-result=void \
8+
// RUN: --shared-libs=%mlir_runner_utils,%mlir_c_runner_utils,%levelzero_runtime | FileCheck %s
99

1010
#map0 = affine_map<(d0, d1, d2, d3, d4) -> (d0, d1, d2, d3, d4)>
1111
#map1 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 * 2 + d5 * 3, d2 * 2 + d6 * 3, d3, d7)>

test/PlaidML/OpTest.Matmul.mlir

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22
// RUN: --runner mlir-cpu-runner -e main \
33
// RUN: --shared-libs=%mlir_runner_utils \
44
// RUN: --entry-point-result=void | FileCheck %s
5-
// RUN-GPU: %python_executable %imex_runner -i %s --pass-pipeline-file=%p/linalg-to-llvm.pp \
6-
// RUN-GPU: --runner mlir-cpu-runner -e main \
7-
// RUN-GPU: --entry-point-result=void \
8-
// RUN-GPU: --shared-libs=%mlir_runner_utils,%levelzero_runtime | FileCheck %s
5+
// RUN: %gpu_skip || %python_executable %imex_runner -i %s --pass-pipeline-file=%p/linalg-to-llvm.pp \
6+
// RUN: --runner mlir-cpu-runner -e main \
7+
// RUN: --entry-point-result=void \
8+
// RUN: --shared-libs=%mlir_runner_utils,%levelzero_runtime | FileCheck %s
99

1010
module @linalg_matmul{
1111
func.func @matmul(%arg0: tensor<5x3xf32>, %arg1: tensor<3x2xf32>) -> (tensor<5x2xf32>) {

0 commit comments

Comments
 (0)