In-place broadcasting into a `VectorOfArray` mishandles arrays with fewer dimensions
维护者通常 3 天内回复
还没有人认领这个 Issue。
评估
调研方向
Start with unpack_voa in src/vector_of_array.jl, which slices every array argument along its last dimension, and with the BroadcastStyle methods for VectorOfArrayStyle that route v .= ... through it. The FastBroadcast extension's fast_materialize! for AbstractVectorOfSArray depends on the same slicing. Done means each case in the issue's table (vector, 1×2, fill(2.0), and 3×2 arguments, with both Vector and SVector inner arrays) matches Array(v) broadcasting, with tests added.
由索引模型根据 Issue 内容生成。
描述
Since v4, a VectorOfArray is a proper AbstractArray (VectorOfArray([[1, 2, 3], [4, 5, 6]]) is a 3×2 array whose columns are the inner arrays), so broadcasting should match broadcasting with Array(v). In-place broadcasting into a VectorOfArray only does so if every array argument has the same number of dimensions as the VectorOfArray. Otherwise it errors or silently returns wrong results:
using RecursiveArrayTools
v = VectorOfArray([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
v .= v .+ [10.0, 20.0, 30.0]
Array(v)
# 3×2 Matrix{Float64}:
# 11.0 24.0
# 12.0 25.0
# 13.0 26.0
#
# but Array(v) .+ [10.0, 20.0, 30.0] is
# 11.0 14.0
# 22.0 25.0
# 33.0 36.0
For the same v (and the same with SVectors as inner arrays):
x |
v .= x .+ 1 |
v .= v .+ x |
|---|---|---|
[10.0, 20.0, 30.0] |
error¹ | wrong result (see above) |
[10.0 20.0] (1×2) |
error¹ | ✓ |
fill(2.0) |
ArgumentError: tuple length should be ≥ 0, got -1 |
same error |
3×2 Matrix |
✓ | ✓ |
¹ DimensionMismatch: destination axes (Base.OneTo(3),) are not compatible with source axes ... with Vectors as inner arrays; MethodError: Cannot convert an object of type ... to an object of type Float64 with SVectors.
Out-of-place broadcasting such as v .+ x gives the correct results for vectors and matrices x (the broadcast style becomes DefaultArrayStyle{2}, so Base handles it). It fails only for fill(2.0), because BroadcastStyle(::VectorOfArrayStyle, ::DefaultArrayStyle{0}) returns the VectorOfArrayStyle, which treats zero-dimensional arrays like scalars.
The cause is unpack_voa (src/vector_of_array.jl):
function unpack_voa(x::AbstractArray{T, N}, i) where {T, N}
return @view x[ntuple(x -> Colon(), N - 1)..., i]
end
It slices every array argument along its own last dimension, as if that dimension indexed the inner arrays. That holds only if ndims(x) equals the number of dimensions of the VectorOfArray. With Base's broadcasting semantics, an argument with fewer dimensions should be broadcast across the inner arrays as a whole (zero-dimensional arrays like scalars), and an argument whose last dimension has size 1 should not be indexed with i.
The FastBroadcast extension relies on the same assumption: fast_materialize!(::Serial, dst::AbstractVectorOfSArray, bc) builds each inner SArray from unpack_voa(bc, i) and needs exactly length(dst.u[i]) elements. For example, with FastBroadcast 1.4.0:
using RecursiveArrayTools, StaticArrays, FastBroadcast
v = VectorOfArray([SVector(1.0, 2.0, 3.0), SVector(4.0, 5.0, 6.0)])
x = [10.0, 20.0, 30.0]
@.. v = x + 1 # ERROR: Generator produced too few elements: Expected exactly 3 elements, but got 1
RecursiveArrayTools 4.5.3, Julia 1.13.1.
🤖 Generated with Claude Code
- 主要语言
- Julia
- 星标
- 234
- 派生
- 77
- 平均合并
- 1 天 16 小时
- 30 天内合并 PR
- 11
环境准备
- 没有 Dockerfile 或 Docker Compose 文件
- 没有 Pull Request 模板
- 阅读贡献指南
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 Pull Request,并在描述里引用这个 Issue 编号。
SciML/RecursiveArrayTools.jl 的其他 Issue
-
难度 1/5 1 小时以内 新手友好度 68/100
SciML/RecursiveArrayTools.jl#255 ·
维护者通常 3 天内回复
-
ArrayPartition of GPU arrays: norm, dot, Vector, any scalar-index; no KernelAbstractions.get_backend未关闭
难度 4/5 3-5 天 新手友好度 45/100
SciML/RecursiveArrayTools.jl#676 ·
维护者通常 3 天内回复
-
难度 3/5 1-2 天 新手友好度 67/100
SciML/RecursiveArrayTools.jl#675 ·
维护者通常 3 天内回复
-
难度 4/5 1-2 天 新手友好度 57/100
SciML/RecursiveArrayTools.jl#674 ·
维护者通常 3 天内回复
-
JET: similar_type(::Any) no-matching-method in RecursiveArrayToolsRaggedArrays (pre-existing QA red)未关闭
难度 4/5 3-5 天 新手友好度 48/100
SciML/RecursiveArrayTools.jl#620 ·
维护者通常 3 天内回复
查看 SciML/RecursiveArrayTools.jl 的全部 Issue
相似的 Issue
-
难度 2/5 1-3 小时 新手友好度 72/100
oscar-system/GAP.jl#1455 ·
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 70/100
SciML/OrdinaryDiffEq.jl#4875 ·
维护者通常 3 天内回复
-
documentation
难度 2/5 1-3 小时 新手友好度 62/100
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 66/100
SciML/NonlinearSolve.jl#1372 ·
维护者通常 1 天内回复
-
Truncated discrete ccdf and logccdf return incorrect values at the lower atom可能已有人在做 关联的 PR 仍在进行中或已合并。 未关闭
难度 2/5 1-3 小时 新手友好度 82/100
JuliaStats/Distributions.jl#2095 ·