Fix OOB in UniformScaling ctor. (#442)

Author: maleadt

src/host/construction.jl

@@ -24,8 +24,8 @@ end
 
 function identity_kernel(ctx::AbstractKernelContext, res::AbstractArray{T}, stride, val) where T
     i = linear_index(ctx)
-    i > stride && return
     ilin = (stride * (i - 1)) + i
+    ilin > length(res) && return
     @inbounds res[ilin] = val
     return
 end

test/testsuite/construction.jl

@@ -178,5 +178,11 @@
             @test eltype(x1) == T
             @test Array(x1) ≈ x
         end
+
+        @testset "JuliaGPU/GPUArrays.jl#439" begin
+            x = AT{Float32}(I, 500, 300)
+            y = Array{Float32}(I, 500, 300)
+            @test Array(x) == y
+        end
     end
 end