improve static performance of perm matrix

QuantumBFS · Feb 26, 2024 · aaf1ff0 · aaf1ff0
1 parent 32f0fe6
commit aaf1ff0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/linalg.jl b/src/linalg.jl
@@ -94,7 +94,7 @@ end
 function Base.:*(A::PermMatrixCSC, B::PermMatrixCSC)
     @assert basetype(A) == basetype(B)
     size(A, 1) == size(B, 1) || throw(DimensionMismatch())
-    basetype(A)(A.perm[B.perm], [B.vals[i] * A.vals[B.perm[i]] for i in 1:size(A, 1)])
+    basetype(A)(A.perm[B.perm], B.vals .* view(A.vals, B.perm))
 end
 
 # to matrix