builtin.module @module_b { | |
// Note the iree.abi.output indicating that the %output argument is storage | |
// for function result 0. | |
func.func @mul(%lhs: tensor<4096xf32>, %rhs: tensor<4096xf32>, | |
%output: tensor<4096xf32> {iree.abi.output = 0 : index}) -> tensor<4096xf32> { | |
%result = arith.mulf %lhs, %rhs : tensor<4096xf32> | |
return %result : tensor<4096xf32> | |
} | |
} |