Val(:H) instruct (#37)

* more instructs * add a test
QuantumBFS · Nov 27, 2019 · 578babc · 578babc
1 parent 95e7b1b
commit 578babc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/instruct.jl b/src/instruct.jl
@@ -16,7 +16,7 @@ end
 
 A list of symbol for specialized gates/operators.
 """
-const SPECIALIZATION_LIST = Symbol[:X, :Y, :Z, :S, :T, :Sdag, :Tdag]
+const SPECIALIZATION_LIST = Symbol[:X, :Y, :Z, :S, :T, :Sdag, :Tdag, :H, :SWAP, :PSWAP, :CPHASE]
 
 function _prepare_instruct(
     state,
@@ -164,6 +164,11 @@ function YaoBase.instruct!(state::AbstractVecOrMat, ::Val{:X}, locs::NTuple{N,In
     return state
 end
 
+function YaoBase.instruct!(state::AbstractVecOrMat, ::Val{:H}, locs::NTuple{N, Int}) where {N}
+    instruct!(state, YaoBase.Const.H, locs)
+end
+
+
 function YaoBase.instruct!(state::AbstractVecOrMat{T}, ::Val{:Y}, locs::NTuple{N,Int}) where {T,N}
     mask = bmask(Int, locs)
     do_mask = bmask(Int, first(locs))

diff --git a/test/instruct.jl b/test/instruct.jl
@@ -149,6 +149,11 @@ end
     @test rot_mat(T, Val(:CPHASE), theta) ≈ rot_mat(T, Diagonal([1, 1, 1, -1]), theta) *
                                             exp(im * theta / 2)
     for ST in [randn(ComplexF64, 1 << 5), randn(ComplexF64, 1 << 5, 10)]
+        @test instruct!(copy(ST), Val(:H), (4,)) ≈ instruct!(
+            copy(ST),
+            Const.H,
+            (4,),
+        )
         for R in [:Rx, :Ry, :Rz]
             @test instruct!(copy(ST), Val(R), (4,), θ) ≈ instruct!(
                 copy(ST),