Fix call to TracedRArray constructor

EnzymeAD · Jul 21, 2024 · 1fbd920 · 1fbd920
1 parent 2b3b277
commit 1fbd920
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Tracing.jl b/src/Tracing.jl
@@ -621,7 +621,7 @@ function broadcast_to_size(arg::Broadcast.Extruded, rsize)
 
     len = length(rsize)
     @assert typeof(len) == Int
-    return TracedRArray{eltype(x),rsize,len}(
+    return TracedRArray{eltype(x),len}(
         (),
         MLIR.IR.result(
             MLIR.Dialects.stablehlo.broadcast_in_dim(
@@ -631,6 +631,7 @@ function broadcast_to_size(arg::Broadcast.Extruded, rsize)
             ),
             1,
         ),
+        rsize,
     )
 end