Skip to content

Commit 3d6f8ec

Browse files
committed
feat: triton working end to end
1 parent adec344 commit 3d6f8ec

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/Ops.jl

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1889,12 +1889,11 @@ function triton_call(
18891889
array_results = ()
18901890
for i in 1:MLIR.IR.nresults(results)
18911891
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-
),
1892+
res = Reactant.TracedRArray{unwrapped_eltype(arg),ndims(arg)}(
1893+
(), MLIR.IR.result(results, i), size(arg)
18971894
)
1895+
copyto!(arg, res)
1896+
array_results = (array_results..., res)
18981897
end
18991898
length(array_results) == 1 && return array_results[1]
19001899
return array_results

0 commit comments

Comments
 (0)