In-place broadcasting into a `VectorOfArray` mishandles arrays with fewer dimensions
Maintainer antworten meist innerhalb von 3 Tagen
Dieses Issue hat noch niemand übernommen.
Bewertung
- Schwierigkeit
- 4/5
- Geschätzter Aufwand
- 3-5 Tage
- Anfängerfreundlichkeit
- 45/100
Rechercherichtung
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.
Vom Indexierungsmodell aus dem Issue-Text verfasst.
Beschreibung
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
- Vorherrschende Sprache
- Julia
- Sterne
- 234
- Forks
- 77
- Ø Merge
- 1 T. 16 Std.
- Gemergte PRs (30 T.)
- 11
Entwicklungsumgebung
- Kein Dockerfile und keine Docker-Compose-Datei
- Keine Pull-Request-Vorlage
- Beitragsleitfaden lesen
Erste Schritte
- Lesen Sie das ganze Issue und danach den Beitragsleitfaden des Projekts.
- Schreiben Sie ins Issue, dass Sie es übernehmen — das erspart doppelte Arbeit.
- Forken Sie das Repository und arbeiten Sie in einem Branch.
- Öffnen Sie einen Pull Request, der die Issue-Nummer nennt.
Mehr aus SciML/RecursiveArrayTools.jl
-
Docs: usage of j vs iOffen
Schwierigkeit 1/5 Unter einer Stunde Anfängerfreundlichkeit 68/100
SciML/RecursiveArrayTools.jl#255 ·
Maintainer antworten meist innerhalb von 3 Tagen
-
ArrayPartition of GPU arrays: norm, dot, Vector, any scalar-index; no KernelAbstractions.get_backendOffen
Schwierigkeit 4/5 3-5 Tage Anfängerfreundlichkeit 45/100
SciML/RecursiveArrayTools.jl#676 ·
Maintainer antworten meist innerhalb von 3 Tagen
-
Schwierigkeit 3/5 1-2 Tage Anfängerfreundlichkeit 67/100
SciML/RecursiveArrayTools.jl#675 ·
Maintainer antworten meist innerhalb von 3 Tagen
-
Schwierigkeit 4/5 1-2 Tage Anfängerfreundlichkeit 57/100
SciML/RecursiveArrayTools.jl#674 ·
Maintainer antworten meist innerhalb von 3 Tagen
-
JET: similar_type(::Any) no-matching-method in RecursiveArrayToolsRaggedArrays (pre-existing QA red)Offen
Schwierigkeit 4/5 3-5 Tage Anfängerfreundlichkeit 48/100
SciML/RecursiveArrayTools.jl#620 ·
Maintainer antworten meist innerhalb von 3 Tagen
Alle Issues in SciML/RecursiveArrayTools.jl
Ähnliche Issues
-
Schwierigkeit 2/5 1-3 Stunden Anfängerfreundlichkeit 72/100
Maintainer antworten meist innerhalb von 1 Tag
-
Schwierigkeit 2/5 1-3 Stunden Anfängerfreundlichkeit 72/100
JuliaGPU/KernelAbstractions.jl#897 ·
Maintainer antworten meist innerhalb von 1 Tag
-
Schwierigkeit 2/5 1-3 Stunden Anfängerfreundlichkeit 72/100
oscar-system/GAP.jl#1455 ·
Maintainer antworten meist innerhalb von 1 Tag
-
Schwierigkeit 2/5 1-3 Stunden Anfängerfreundlichkeit 65/100
JuliaAI/MLJTransforms.jl#85 ·
-
Schwierigkeit 2/5 1-3 Stunden Anfängerfreundlichkeit 70/100
SciML/OrdinaryDiffEq.jl#4875 ·
Maintainer antworten meist innerhalb von 4 Tagen