From b0dd7a75187f2c0cc0e97bbf1281e13753ba1174 Mon Sep 17 00:00:00 2001 From: "colin.liang" Date: Thu, 15 Jul 2021 16:26:17 +0800 Subject: [PATCH] add mul in test --- tests/test.mlir | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test.mlir b/tests/test.mlir index b8bcebd..7e2ff26 100644 --- a/tests/test.mlir +++ b/tests/test.mlir @@ -50,9 +50,10 @@ func @main(%1 : memref<*xf32>,%2 : memref<*xf32> ) -> memref<*xf32> { %in1 = memref.tensor_load %ranked1 : memref<6xf32> %in2 = memref.tensor_load %ranked2 : memref<6xf32> - %add = "mhlo.add"(%in1, %in2) {name = "add.3"} : (tensor<6xf32>, tensor<6xf32>) -> tensor<6xf32> + %add = "mhlo.multiply"(%in1, %in2) {name = "mul.3"} : (tensor<6xf32>, tensor<6xf32>) -> tensor<6xf32> + %mul = "mhlo.add"(%in1, %add) {name = "add.3"} : (tensor<6xf32>, tensor<6xf32>) -> tensor<6xf32> - %output = memref.buffer_cast %add : memref<6xf32> + %output = memref.buffer_cast %mul : memref<6xf32> %unranked_output = memref.cast %output : memref<6xf32> to memref<*xf32> call @print_memref_f32(%1) : (memref<*xf32>) -> ()