Skip to content

Commit f442b35

Browse files
committed
Fix collection type passed to stablehlo.concatenate
1 parent 296b0ca commit f442b35

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/TracedRArray.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -779,7 +779,7 @@ function Base._cat_t(dims, ::Type{T}, X::TracedRArray...) where {T}
779779
MLIR.IR.result(
780780
# TODO maybe we should do some conversion?
781781
MLIR.Dialects.stablehlo.concatenate(
782-
get_mlir_data.(X);
782+
collect(get_mlir_data.(X));
783783
result_0=MLIR.IR.TensorType(shape, MLIR.IR.Type(RT)),
784784
dimension=dims - 1, # stablehlo expects this to be zero-indexed
785785
),

0 commit comments

Comments
 (0)