We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent adec344 commit 3d6f8ecCopy full SHA for 3d6f8ec
src/Ops.jl
@@ -1889,12 +1889,11 @@ function triton_call(
1889
array_results = ()
1890
for i in 1:MLIR.IR.nresults(results)
1891
arg = args[output_to_arg[i]]
1892
- array_results = (
1893
- array_results...,
1894
- Reactant.TracedRArray{unwrapped_eltype(arg),ndims(arg)}(
1895
- (), MLIR.IR.result(results, i), size(arg)
1896
- ),
+ res = Reactant.TracedRArray{unwrapped_eltype(arg),ndims(arg)}(
+ (), MLIR.IR.result(results, i), size(arg)
1897
)
+ copyto!(arg, res)
+ array_results = (array_results..., res)
1898
end
1899
length(array_results) == 1 && return array_results[1]
1900
return array_results
0 commit comments