Add tests

JuliaGPU · Dec 19, 2024 · 1bb9247 · 1bb9247
1 parent ef02e98
commit 1bb9247
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/core/cudadrv.jl b/test/core/cudadrv.jl
@@ -689,6 +689,20 @@ if attribute(device(), CUDA.DEVICE_ATTRIBUTE_HOST_REGISTER_SUPPORTED) != 0
     dA = CUDA.rand(UInt8, 512)
     copyto!(dA, hA)
     copyto!(hA, dA)
+
+    # memory copies with resized pinned memory (used to fail with CUDA_ERROR_INVALID_VALUE)
+    dA = rand(Float32, 100)
+    hA = Array(dA)
+    @test !CUDA.is_pinned(pointer(hA))
+    for n ∈ 100:2000
+        resize!(dA, n)
+        resize!(hA, n)
+        dA .= n
+        CUDA.pin(hA)
+        @test CUDA.is_pinned(pointer(hA))
+        copyto!(hA, dA)
+        copyto!(dA, hA)
+    end
 end
 
 end