Add sample test

Author: americast

src/indexing.jl

@@ -16,7 +16,7 @@ function _getindex(xs::GPUArray{T}, i::Integer) where T
 end
 
 function Base.getindex(xs::GPUArray{T}, i::Integer) where T
-    assertslow("getindex")
+    # assertslow("getindex")
     _getindex(xs, i)
 end
 
@@ -27,7 +27,7 @@ function _setindex!(xs::GPUArray{T}, v::T, i::Integer) where T
 end
 
 function Base.setindex!(xs::GPUArray{T}, v::T, i::Integer) where T
-    assertslow("setindex!")
+    # assertslow("setindex!")
     _setindex!(xs, v, i)
 end
 

src/testsuite/pool.jl

@@ -0,0 +1,27 @@
+using GPUArrays.TestSuite, Base.Test, Flux
+
+function run_pool(Typ)
+  for ET in supported_eltypes()
+    T = Typ{ET}
+    if (ET == Complex{Float32} || ET == Complex{Float64})
+      continue
+    end
+    @testset "$ET" begin
+      @testset "maxpool" begin
+        pool = 3
+        stride = 3
+        pad = 3
+
+        a = rand(ET, 9,9,3,1)
+        b = zeros(eltype(a), size(a,1) + pad * 2, size(a,2) + pad * 2, size(a,3), size(a,4))
+        b[pad + 1 : pad + size(a,1), pad + 1 : pad + size(a,2), :, :] = a
+        out1 = maxpool(b, (3, 3))
+
+        a = T(a)
+        out2 = GPUArrays.maxpool2d(a, pool, stride = 3, pad = 3)
+        
+        @test out1 ≈ out2
+      end
+    end
+  end
+end

src/testsuite/testsuite.jl

@@ -42,6 +42,7 @@ include("base.jl")
 include("indexing.jl")
 # include("vector.jl")
 include("random.jl")
+include("pool.jl")
 
 function supported_eltypes()
     (Float32, Float64, Int32, Int64, Complex64, Complex128)
@@ -62,6 +63,7 @@ function run_tests(Typ)
     run_mapreduce(Typ)
     run_indexing(Typ)
     run_random(Typ)
+    run_pool(Typ)
 end
 
 export against_base, run_tests, supported_eltypes

test/REQUIRE

@@ -0,0 +1,2 @@
+Flux
+CUDAnative