Infer arraytype for adjoint and transpose

JuliaMath · Sep 7, 2024 · d4d8caf · d4d8caf
1 parent 8b84e97
commit d4d8caf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/AbstractNFFTs/src/derived.jl b/AbstractNFFTs/src/derived.jl
@@ -23,6 +23,9 @@ $(planfunc)(k::arrT, N::Union{Integer,NTuple{D,Int}}, args...; kargs...) where {
 $(planfunc)(k::arrT, y::AbstractArray, args...; kargs...) where {arrT <: AbstractArray} =
     $(planfunc)(strip_type_parameters(arrT), k, y, args...; kargs...)
 
+$(planfunc)(k::arrL, args...; kargs...) where {T, arrT <: AbstractArray{T}, arrL <: Union{Adjoint{T, arrT}, Transpose{T, arrT}}} =
+    $(planfunc)(strip_type_parameters(arrT), k, y, args...; kargs...)
+
 # The follow convert 1D parameters into the format required by the plan
 
 $(planfunc)(Q::Type, k::AbstractVector, N::Integer, rest...; kwargs...)  =