Renable CuArray construction test

compintell · Jan 8, 2025 · 51357db · 51357db
1 parent 69750f5
commit 51357db
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/test/ext/cuda/cuda.jl b/test/ext/cuda/cuda.jl
@@ -3,7 +3,7 @@ Pkg.activate(@__DIR__)
 Pkg.develop(; path=joinpath(@__DIR__, "..", "..", ".."))
 
 using AllocCheck, CUDA, JET, Mooncake, StableRNGs, Test
-using Mooncake.TestUtils: test_tangent_consistency, test_fwds_rvs_data
+using Mooncake.TestUtils: test_tangent_consistency, test_fwds_rvs_data, test_rule
 
 @testset "cuda" begin
 
@@ -12,14 +12,14 @@ using Mooncake.TestUtils: test_tangent_consistency, test_fwds_rvs_data
     test_tangent_consistency(StableRNG(123456), p; interface_only=false)
     test_fwds_rvs_data(StableRNG(123456), p)
 
-    # # Check we can instantiate a CuArray.
-    # test_rule(
-    #     StableRNG(123456),
-    #     CuArray{Float32,1,CUDA.DeviceMemory},
-    #     undef,
-    #     256;
-    #     interface_only=true,
-    #     is_primitive=true,
-    #     debug_mode=true,
-    # )
+    # Check we can instantiate a CuArray.
+    test_rule(
+        StableRNG(123456),
+        CuArray{Float32,1,CUDA.DeviceMemory},
+        undef,
+        256;
+        interface_only=true,
+        is_primitive=true,
+        debug_mode=true,
+    )
 end