Skip to content

Commit 9e930e4

Browse files
committed
Fix fusedims ambiguity
1 parent a9b66d6 commit 9e930e4

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

src/fusedims.jl

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,17 @@ end
2929
() = Base.OneTo(1)
3030

3131
# Overload this version for most arrays
32-
function fusedims(a::AbstractArray, axes::AbstractUnitRange...)
33-
return fusedims(FusionStyle(a), a, axes...)
32+
function fusedims(a::AbstractArray, ax::AbstractUnitRange, axes::AbstractUnitRange...)
33+
return fusedims(FusionStyle(a), a, ax, axes...)
3434
end
3535

3636
# Overload this version for fusion tensors, array maps, etc.
37-
function fusedims(a::AbstractArray, axesblocks::Tuple{Vararg{AbstractUnitRange}}...)
38-
return fusedims(a, flatten_tuples(axesblocks)...)
37+
function fusedims(
38+
a::AbstractArray,
39+
axb::Tuple{Vararg{AbstractUnitRange}},
40+
axesblocks::Tuple{Vararg{AbstractUnitRange}}...,
41+
)
42+
return fusedims(a, flatten_tuples((axb, axesblocks...))...)
3943
end
4044

4145
# Fix ambiguity issue

0 commit comments

Comments
 (0)