fix

wsmoses · wsmoses · commit 37a05512de1b · 2025-07-23T17:24:35.000-05:00
diff --git a/src/ConcreteRArray.jl b/src/ConcreteRArray.jl
@@ -387,7 +387,7 @@ function Base.similar(a::ConcretePJRTArray{T, N, D, Sh}, ::Type{S}=T, dims::Dims
         )
         ntuple(Val(D)) do i
             Base.@_inline_meta
-            Base.similar(a.data[i], S, Dims(device_to_array_slices[i]))
+	    Base.similar(a.data[i], S, Dims(size(device_to_array_slices[i])))
         end
     end
 

Original file line number	Diff line number	Diff line change
`@@ -387,7 +387,7 @@ function Base.similar(a::ConcretePJRTArray{T, N, D, Sh}, ::Type{S}=T, dims::Dims`
`387`	`387`	`)`
`388`	`388`	`ntuple(Val(D)) do i`
`389`	`389`	`Base.@_inline_meta`
`390`		`- Base.similar(a.data[i], S, Dims(device_to_array_slices[i]))`
	`390`	`+ Base.similar(a.data[i], S, Dims(size(device_to_array_slices[i])))`
`391`	`391`	`end`
`392`	`392`	`end`
`393`	`393`