test: object not copied

Author: avik-pal

lib/MLDataDevices/test/amdgpu_tests.jl

@@ -124,6 +124,12 @@ using FillArrays, Zygote  # Extensions
         return_val(x) = Val(get_device_type(x))  # If it is a compile time constant then type inference will work
         @test @inferred(return_val(ps)) isa Val{parameterless_type(typeof(device))}
     end
+
+    @testset "Issue #1129: no new object" begin
+        x = rand(Float32, 10, 10) |> device
+        y = x |> device
+        @test x === y
+    end
 end
 
 @testset "Functions" begin

lib/MLDataDevices/test/cuda_tests.jl

@@ -149,6 +149,12 @@ using FillArrays, Zygote  # Extensions
         return_val2(x) = Val(get_device(x))
         @test_throws ErrorException @inferred(return_val2(ps))
     end
+
+    @testset "Issue #1129: no new object" begin
+        x = rand(Float32, 10, 10) |> device
+        y = x |> device
+        @test x === y
+    end
 end
 
 @testset "Functions" begin

lib/MLDataDevices/test/metal_tests.jl

@@ -113,6 +113,12 @@ using FillArrays, Zygote  # Extensions
         return_val2(x) = Val(get_device(x))
         @test @inferred(return_val2(ps)) isa Val{get_device(x)}
     end
+
+    @testset "Issue #1129: no new object" begin
+        x = rand(Float32, 10, 10) |> device
+        y = x |> device
+        @test x === y
+    end
 end
 
 @testset "Functions" begin

lib/MLDataDevices/test/oneapi_tests.jl

@@ -113,6 +113,12 @@ using FillArrays, Zygote  # Extensions
         return_val2(x) = Val(get_device(x))
         @test @inferred(return_val2(ps)) isa Val{get_device(x)}
     end
+
+    @testset "Issue #1129: no new object" begin
+        x = rand(Float32, 10, 10) |> device
+        y = x |> device
+        @test x === y
+    end
 end
 
 @testset "Functions" begin

lib/MLDataDevices/test/xla_tests.jl

@@ -113,6 +113,12 @@ using FillArrays, Zygote  # Extensions
         return_val2(x) = Val(get_device(x))
         @test_throws TypeError @inferred(return_val2(ps))
     end
+
+    @testset "Issue #1129: no new object" begin
+        x = rand(Float32, 10, 10) |> device
+        y = x |> device
+        @test x === y
+    end
 end
 
 @testset "Functions" begin