conversion test

2022-06-08 16:19:15 +08:00
parent 26fcc12afd
commit 49d1821149
1 changed files with 9 additions and 0 deletions
--- a/test/Conversion/ops.mlir
+++ b/test/Conversion/ops.mlir
@@ -0,0 +1,9 @@
+// RUN triton-opt %s -convert-triton-to-tritongpu
+
+func @ops() {
+  %a = arith.constant dense<1.00e+00> : tensor<128x32xf16>
+  %b = arith.constant dense<2.00e+00> : tensor<32x128xf16>
+  %c = arith.constant dense<3.00e+00> : tensor<128x128xf32>
+  %0 = tt.dot %a, %b, %c {allowTF32 = true} : tensor<128x32xf16> * tensor<32x128xf16> -> tensor<128x128xf32>
+  return
+}