From 33f30577fb32ac05af827f2fcecc23818eae5c18 Mon Sep 17 00:00:00 2001
From: Jun Tian <tianjun.cpp@gmail.com>
Date: Mon, 21 Sep 2020 17:50:24 +0800
Subject: [PATCH] minor enhancement

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

diff --git a/src/utils/device.jl b/src/utils/device.jl
index a4d727c..cf39f8e 100644
--- a/src/utils/device.jl
+++ b/src/utils/device.jl
@@ -16,14 +16,13 @@ send_to_device(::Val{:gpu}, x) = Flux.fmap(a -> adapt(CuArray{Float32}, a), x)
 send_to_device(
     ::Val{:gpu},
     x::Union{
-        SubArray{<:Any,<:Any,<:CircularArrayBuffer},
+        SubArray{<:Any,<:Any,<:Union{CircularArrayBuffer, ElasticArray}},
         Base.ReshapedArray{<:Any,<:Any,<:SubArray{<:Any,<:Any,<:CircularArrayBuffer}},
         SubArray{
             <:Any,
             <:Any,
-            <:Base.ReshapedArray{<:Any,<:Any,<:SubArray{<:Any,<:Any,<:CircularArrayBuffer}},
+            <:Base.ReshapedArray{<:Any,<:Any,<:SubArray{<:Any,<:Any,<:Union{CircularArrayBuffer, ElasticArray}}},
         },
-        SubArray{<:Any,<:Any,<:ElasticArray,<:Any,<:Any},
         ElasticArray,
     },
 ) = CuArray(x)