Skip to content

Commit 0e5fa4a

Browse files
committed
add batch test run line
1 parent 3b20bda commit 0e5fa4a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

enzyme/test/MLIR/Batch/cachefunction.mlir

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
// RUN: %eopt -enzyme-batch %s | FileCheck %s
2+
13
module {
24
func.func private @f(%arg0: tensor<16xf32>, %arg1: tensor<16xf32>) -> tensor<16xf32> {
35
return %arg0 : tensor<16xf32>
@@ -16,4 +18,4 @@ module {
1618
// CHECK-NEXT: }
1719
// CHECK: func.func private @batched_f(%[[arg0:.+]]: tensor<4x16xf32>, %[[arg1:.+]]: tensor<4x16xf32>) -> tensor<4x16xf32> {
1820
// CHECK-NEXT: return %[[arg0]] : tensor<4x16xf32>
19-
// CHECK-NEXT: }
21+
// CHECK-NEXT: }

0 commit comments

Comments
 (0)