fix batched_kron!

QuantumBFS · Aug 30, 2019 · 17ec5df · 17ec5df
1 parent 86d74bd
commit 17ec5df
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/utils.jl b/src/utils.jl
@@ -221,3 +221,9 @@ end
     end
     state
 end
+
+#### Yao Base patch ####
+using YaoBase
+function YaoBase.batched_kron!(C::Array{T, 3}, A::AbstractArray{T1, 3}, B::AbstractArray{T2, 3}) where {T, T1, T2}
+    YaoBase.batched_kron!(C, convert(Array{T,3}, A), convert(Array{T,3}, B))
+end