| Ben Vanik | dc6f0cd | 2023-11-20 21:36:58 -0800 | [diff] [blame] | 1 | builtin.module @module_b { |
| 2 | // Note the iree.abi.output indicating that the %output argument is storage | ||||
| 3 | // for function result 0. | ||||
| 4 | func.func @mul(%lhs: tensor<4096xf32>, %rhs: tensor<4096xf32>, | ||||
| 5 | %output: tensor<4096xf32> {iree.abi.output = 0 : index}) -> tensor<4096xf32> { | ||||
| 6 | %result = arith.mulf %lhs, %rhs : tensor<4096xf32> | ||||
| 7 | return %result : tensor<4096xf32> | ||||
| 8 | } | ||||
| 9 | } | ||||