colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add mul in test

This commit is contained in:
colin.liang 2021-07-15 16:26:17 +08:00
parent e62fa9311f
commit b0dd7a7518
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test.mlir
View File

@ -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>) -> ()