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>
  }
}