Merge pull request #231 from janEbert/clamp

Add Base.clamp!

Author: maleadt

src/GPUArrays.jl

@@ -30,6 +30,7 @@ include("host/indexing.jl")
 include("host/broadcast.jl")
 include("host/mapreduce.jl")
 include("host/linalg.jl")
+include("host/math.jl")
 include("host/random.jl")
 include("host/quirks.jl")
 

src/host/math.jl

@@ -0,0 +1,11 @@
+import Base.clamp!
+
+function Base.clamp!(A::AbstractGPUArray, low, high)
+    function kernel(state, A, low, high)
+        I = @cartesianidx A state
+        A[I...] = clamp(A[I...], low, high)
+        return
+    end
+    gpu_call(kernel, A, low, high)
+    return A
+end

test/testsuite.jl

@@ -37,6 +37,7 @@ include("testsuite/vector.jl")
 include("testsuite/mapreduce.jl")
 include("testsuite/broadcasting.jl")
 include("testsuite/linalg.jl")
+include("testsuite/math.jl")
 include("testsuite/fft.jl")
 include("testsuite/random.jl")
 
@@ -53,6 +54,7 @@ function test(AT::Type{<:AbstractGPUArray})
     TestSuite.test_mapreduce(AT)
     TestSuite.test_broadcasting(AT)
     TestSuite.test_linalg(AT)
+    TestSuite.test_math(AT)
     TestSuite.test_fft(AT)
     TestSuite.test_random(AT)
 end

test/testsuite/math.jl

@@ -0,0 +1,20 @@
+function test_math(AT)
+    @testset "math functionality" begin
+        for ET in supported_eltypes()
+            # Skip complex numbers
+            ET in (Complex, ComplexF32, ComplexF64) && continue
+
+            T = AT{ET}
+            @testset "$ET" begin
+                range = ET <: Integer ? (ET(-2):ET(2)) : ET
+                low = ET(-1)
+                high = ET(1)
+                @testset "clamp!" begin
+                    for N in (2, 10)
+                        @test compare(x -> clamp!(x, low, high), AT, rand(range, N, N))
+                    end
+                end
+            end
+        end
+    end
+end