blob: a23365c90fa2a69a68c44449b7f3e392f6222b93 [file] [log] [blame]
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>
}
}