From 167d500abd5ff2608c56bf9ed6c03ceead9863cf Mon Sep 17 00:00:00 2001
From: Joe Greener <jgreener@hotmail.co.uk>
Date: Wed, 3 Aug 2022 15:53:28 +0100
Subject: [PATCH] update for changes on master

---
 src/lib/broadcast.jl | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/lib/broadcast.jl b/src/lib/broadcast.jl
index 8d6a93894..086fc6d13 100644
--- a/src/lib/broadcast.jl
+++ b/src/lib/broadcast.jl
@@ -286,11 +286,10 @@ using GPUArraysCore  # replaces @require CUDA block, weird indenting to preserve
 
   pull_block_vert(sz, Δ::AbstractGPUArray, A::Number) = @allowscalar Δ[sz]
 
-  ∇getindex(x::CUDA.CuArray, inds::Tuple{AbstractArray{<:Integer}}) = dy -> begin
+  ∇getindex(x::T, inds::Tuple{AbstractArray{<:Integer}}) where {T <: AbstractGPUArray} = dy -> begin
     inds1_cpu = Array(inds[1])
     dx = zeros(eltype(dy), length(x))
     dxv = view(dx, inds1_cpu)
     dxv .= accum.(dxv, _droplike(Array(dy), dxv))
-    return _project(x, CUDA.CuArray(dx)), nothing
+    return _project(x, T(dx)), nothing
   end
-end