From 5c15d7ee36099f157516cb7eb64d8a7a2b1d2cea Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 23 Sep 2020 00:10:08 +0800 Subject: [PATCH] updated send_to_device for ElasticArray. --- src/utils/device.jl | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/utils/device.jl b/src/utils/device.jl index 891b19f..627095b 100644 --- a/src/utils/device.jl +++ b/src/utils/device.jl @@ -17,7 +17,12 @@ send_to_device( ::Val{:gpu}, x::Union{ SubArray{<:Any,<:Any,<:Union{CircularArrayBuffer,ElasticArray}}, - Base.ReshapedArray{<:Any,<:Any,<:SubArray{<:Any,<:Any,<:CircularArrayBuffer}}, + Base.ReshapedArray{ + <:Any, + <:Any, + <:SubArray{<:Any,<:Any,<:Union{CircularArrayBuffer,ElasticArray}}, + }, + Base.ReshapedArray{<:Any,<:Any,<:Union{CircularArrayBuffer,ElasticArray}}, SubArray{ <:Any, <:Any,