Skip to content

Commit a85fb9d

Browse files
committed
add test case
1 parent 5466610 commit a85fb9d

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
// RUN: gc-opt %s -decompose-aggregated-ops | FileCheck %s
2+
3+
// CHECK-LABEL: softmax
4+
func.func @softmax(%arg0: tensor<2x2x2x2xf32>) -> tensor<2x2x2x2xf32> {
5+
%0 = tensor.empty() : tensor<2x2x2x2xf32>
6+
// CHECK-NOT: linalg.softmax
7+
// CHECK-COUNT-4: linalg.generic
8+
%1 = linalg.softmax dimension(3)
9+
ins(%arg0 : tensor<2x2x2x2xf32>) outs(%0 : tensor<2x2x2x2xf32>) -> tensor<2x2x2x2xf32>
10+
return %1 : tensor<2x2x2x2xf32>
11+
}

0 commit comments

Comments
 (0)