Update src/gather.jl

Co-authored-by: Peter <[email protected]>

Author: yuehhua

src/gather.jl

@@ -53,7 +53,7 @@ function checkbounds_src(src, dims::Union{Int, Val}, ::Type{<:Any})
     return i -> checkbounds(Bool, src, ntuple(x -> Colon(), dims)..., i...)
 end
 
-function checkbounds_src(src, dims::Int, ::Type{<:CartesianIndex})
+function checkbounds_src(src, dims::Union{Int, Val}, ::Type{<:CartesianIndex})
     return i -> checkbounds(Bool, src, ntuple(x -> Colon(), dims)..., i)
 end