reduce JET errors

QuantumSavory · Sep 19, 2024 · 2ed7a3e · 2ed7a3e
1 parent 05491d9
commit 2ed7a3e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/fastmemlayout.jl b/src/fastmemlayout.jl
@@ -16,8 +16,8 @@ See also: [`fastrow`](@ref)"""
 function fastcolumn end
 
 fastrow(t::Tableau{Tₚᵥ,Tₘ}) where {Tₚᵥ, Tₘ} = t
-fastrow(t::Tableau{Tₚᵥ,Tₘ}) where {Tₚᵥ, Tₘ<:Adjoint} = Tableau(t.phases, t.nqubits, collect(t.xzs))
-fastcolumn(t::Tableau{Tₚᵥ,Tₘ}) where {Tₚᵥ, Tₘ} = Tableau(t.phases, t.nqubits, collect(t.xzs')')
+fastrow(t::Tableau{Tₚᵥ,Tₘ}) where {Tₚᵥ, Tₘ<:Adjoint} = Tableau(t.phases, t.nqubits, Matrix(collect(t.xzs)))
+fastcolumn(t::Tableau{Tₚᵥ,Tₘ}) where {Tₚᵥ, Tₘ} = Tableau(t.phases, t.nqubits, Matrix(collect(t.xzs')'))
 fastcolumn(t::Tableau{Tₚᵥ,Tₘ}) where {Tₚᵥ, Tₘ<:Adjoint} = t
 
 fastrow(s::Stabilizer) = Stabilizer(fastrow(s.tab))

diff --git a/src/petrajectory.jl b/src/petrajectory.jl
@@ -34,7 +34,7 @@ function petrajectory(state, circuit; branch_weight=1.0, current_order=0, max_or
         if status==continue_stat # TODO is the copy below necessary?
             out_probs = petrajectory(copy(newstate), rest_of_circuit,
                 branch_weight=branch_weight*prob, current_order=current_order+order, max_order=max_order)
-            status_probs .+= out_probs
+            map!((x,y)-> x+y, status_probs, status_probs, out_probs)
         else
             status_probs[status.status] += prob*branch_weight
         end

diff --git a/test/test_jet.jl b/test/test_jet.jl
@@ -23,5 +23,5 @@
 
     @show rep
     @test_broken length(JET.get_reports(rep)) == 0
-    @test length(JET.get_reports(rep)) <= 11
+    @test length(JET.get_reports(rep)) <= 9
 end