fix up copy

Author: sbrantq

src/ProbProg.jl

@@ -47,8 +47,13 @@ function addSampleToTraceLowered(
         trace[symbol] = unsafe_load(reinterpret(Ptr{element_type}, sample_ptr))
     else
         trace[symbol] = Base.deepcopy(
-            unsafe_wrap(
-                Array{element_type}, reinterpret(Ptr{element_type}, sample_ptr), prod(shape)
+            reshape(
+                unsafe_wrap(
+                    Array{element_type},
+                    reinterpret(Ptr{element_type}, sample_ptr),
+                    prod(shape),
+                ),
+                shape,
             ),
         )
     end

test/probprog/simulate.jl

@@ -8,15 +8,14 @@ function simulate_model(seed, μ, σ, shape)
     function model(seed, μ, σ, shape)
         rng = Random.default_rng()
         Random.seed!(rng, seed)
-        s = ProbProg.sample!(normal, rng, μ, σ, shape; symbol = :s)
-        t = ProbProg.sample!(normal, rng, s, σ, shape; symbol = :t)
+        s = ProbProg.sample!(normal, rng, μ, σ, shape; symbol=:s)
+        t = ProbProg.sample!(normal, rng, s, σ, shape; symbol=:t)
         return t
     end
 
     return ProbProg.simulate!(model, seed, μ, σ, shape)
 end
 
-
 @testset "Simulate" begin
     @testset "normal_hlo" begin
         shape = (10000,)
@@ -36,7 +35,7 @@ end
     end
 
     @testset "normal_simulate" begin
-        shape = (10,)
+        shape = (3, 3, 3)
         seed = Reactant.to_rarray(UInt64[1, 4])
         μ = Reactant.ConcreteRArray(0.0)
         σ = Reactant.ConcreteRArray(1.0)