Skip to content

Commit 8b4cbf3

Browse files
committed
Fix unbound args in multi-tree eval
1 parent 08df9a3 commit 8b4cbf3

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

ext/DynamicExpressionsCUDAExt.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ function eval_tree_array(
2929
end
3030

3131
function eval_tree_array(
32-
trees::Union{NTuple{M,N} where M,AbstractVector{N}},
32+
trees::Union{Tuple{N,Vararg{N}},AbstractVector{N}},
3333
gcX::MaybeCuArray{T,2},
3434
operators::OperatorEnum;
3535
buffer=nothing,

src/AsArray.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ using ..EquationModule: AbstractExpressionNode, tree_mapreduce, count_nodes
44

55
function as_array(
66
::Type{I},
7-
trees::Union{NTuple{M,N} where M,AbstractVector{N}};
7+
trees::Union{Tuple{N,Vararg{N}},AbstractVector{N}};
88
buffer::Union{AbstractArray,Nothing}=nothing,
99
) where {T,N<:AbstractExpressionNode{T},I}
1010
each_num_nodes = (t -> count_nodes(t; break_sharing=Val(true))).(trees)

src/EvaluateEquation.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ function eval_tree_array(
9898
return eval_tree_array(tree, cX, operators; turbo, bumper)
9999
end
100100
function eval_tree_array(
101-
trees::Union{NTuple{M,N} where M,AbstractArray{N}},
101+
trees::Union{Tuple{N,Vararg{N}},AbstractVector{N}},
102102
cX::AbstractMatrix{T},
103103
operators::OperatorEnum;
104104
kws...,

0 commit comments

Comments
 (0)