Use fill for res in GradOp

Author: nHackel

ext/LinearOperatorGPUArraysExt/GradientOp.jl

@@ -14,7 +14,7 @@ function LinearOperatorCollection.grad!(res::vecT, img::vecT, shape, dim) where
 end
 
 # adjoint of directional gradients
-function LinearOperatorCollection.grad_t!(res::vecT, g::vecT, shape::NTuple{N,Int64}, dim::Int64) where {vecT <: AbstractGPUVector, N}
+function LinearOperatorCollection.grad_t!(res::vecT, g::vecT, shape::NTuple{N,Int64}, dim::Int64) where {T, vecT <: AbstractGPUVector{T}, N}
   δ = zeros(Int, length(shape))
   δ[dim] = 1
   δ = Tuple(δ)
@@ -23,7 +23,7 @@ function LinearOperatorCollection.grad_t!(res::vecT, g::vecT, shape::NTuple{N,In
   res_ = reshape(res,shape)
   g_ = reshape(g, shape .- δ)
 
-  res_ .= 0
+  fill!(res, zero(T))
   gpu_call(res_, g_, di, elements = length(g)) do ctx, res_k, g_k, di_k
     idx = @cartesianidx(g_k)
     @inbounds res_k[idx]  = g_k[idx]