mirror of
https://github.com/kalmarek/PropertyT.jl.git
synced 2024-12-24 18:10:29 +01:00
optimized version of compue_SOS
using accumulator matrix to allow auto-vectorisation
This commit is contained in:
parent
2602d0a181
commit
60bdbd0e74
@ -3,19 +3,50 @@ using IntervalArithmetic
|
|||||||
IntervalArithmetic.setrounding(Interval, :tight)
|
IntervalArithmetic.setrounding(Interval, :tight)
|
||||||
IntervalArithmetic.setformat(sigfigs=12)
|
IntervalArithmetic.setformat(sigfigs=12)
|
||||||
|
|
||||||
function compute_SOS(pm::Array{I,2}, Q) where I<:Integer
|
function fma_SOS_thr!(result::AbstractVector{T}, pm::AbstractMatrix{<:Integer},
|
||||||
result = zeros(eltype(Q), maximum(pm));
|
Q::AbstractMatrix{T}, acc_matrix=zeros(T, size(pm)...)) where T
|
||||||
for i in 1:size(Q,2)
|
|
||||||
GroupRings.fmac!(result, view(Q,:,i), view(Q,:,i), pm)
|
s1, s2 = size(pm)
|
||||||
|
|
||||||
|
@inbounds for k in 1:s2
|
||||||
|
let k=k, s1=s1, s2=s2, Q=Q, acc_matrix=acc_matrix
|
||||||
|
Threads.@threads for j in 1:s2
|
||||||
|
for i in 1:s1
|
||||||
|
@inbounds acc_matrix[i,j] = muladd(Q[i, k], Q[j, k], acc_matrix[i,j])
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
@inbounds for j in 1:s2
|
||||||
|
for i in 1:s1
|
||||||
|
result[pm[i,j]] += acc_matrix[i,j]
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
return result
|
return result
|
||||||
end
|
end
|
||||||
|
|
||||||
function compute_SOS(RG::GroupRing, Q::AbstractArray)
|
function compute_SOS(pm::AbstractMatrix{<:Integer}, Q::AbstractMatrix)
|
||||||
|
result = zeros(eltype(Q), maximum(pm));
|
||||||
|
return fma_SOS_thr!(result, pm, Q)
|
||||||
|
end
|
||||||
|
|
||||||
|
function compute_SOS(RG::GroupRing, Q::AbstractMatrix{<:Real})
|
||||||
result = compute_SOS(RG.pm, Q)
|
result = compute_SOS(RG.pm, Q)
|
||||||
return GroupRingElem(result, RG)
|
return GroupRingElem(result, RG)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
function compute_SOS_square(RG::GroupRing, Q::AbstractMatrix{<:Real})
|
||||||
|
result = zeros(eltype(Q), maximum(RG.pm));
|
||||||
|
|
||||||
|
for i in 1:size(Q,2)
|
||||||
|
GroupRings.fmac!(result, view(Q,:,i), view(Q,:,i), RG.pm)
|
||||||
|
end
|
||||||
|
|
||||||
|
return GroupRingElem(result, RG)
|
||||||
|
end
|
||||||
|
|
||||||
function augIdproj(Q::AbstractMatrix{T}) where {T<:Real}
|
function augIdproj(Q::AbstractMatrix{T}) where {T<:Real}
|
||||||
result = zeros(size(Q))
|
result = zeros(size(Q))
|
||||||
l = size(Q, 2)
|
l = size(Q, 2)
|
||||||
|
Loading…
Reference in New Issue
Block a user