kalmar/PropertyT.jl
1
0
Fork 0
mirror of https://github.com/kalmarek/PropertyT.jl.git synced 2024-12-25 02:15:29 +01:00
Code Issues Releases Wiki Activity

optimized version of compue_SOS

Browse Source 
using accumulator matrix to allow auto-vectorisation
This commit is contained in:
kalmarek 2019-03-15 08:13:38 +01:00
parent 2602d0a181
commit 60bdbd0e74
No known key found for this signature in database
GPG Key ID: 8BF1A3855328FC15
1 changed files with 36 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/checksolution.jl
View File

@ -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)