Thread.@threads compute_SOS (better precision!)

This commit is contained in:
kalmarek 2019-03-12 15:47:31 +01:00
parent d3ad8b9c6b
commit ad1b11f10d
No known key found for this signature in database
GPG Key ID: 8BF1A3855328FC15
1 changed files with 20 additions and 6 deletions

View File

@ -3,20 +3,34 @@ using IntervalArithmetic
IntervalArithmetic.setrounding(Interval, :tight)
IntervalArithmetic.setformat(sigfigs=12)
function compute_SOS(pm::Array{I,2}, Q) where I<:Integer
result = zeros(eltype(Q), maximum(pm));
for i in 1:size(Q,2)
GroupRings.fmac!(result, view(Q,:,i), view(Q,:,i), pm)
function compute_SOS(pm::AbstractMatrix{<:Integer}, Q::AbstractMatrix{<:Real})
thr_count = Threads.nthreads()
d, r = divrem(size(Q,2), thr_count)
batch_result = [zeros(eltype(Q), maximum(pm)) for _ in 1:thr_count]
Threads.@threads for k in 1:Threads.nthreads()
for i in 1:d
idx = d*(k-1)+i
GroupRings.fmac!(batch_result[k], view(Q,:,idx), view(Q,:,idx), pm)
end
end
result = sum(batch_result)
for idx in thr_count*d+1:(thr_count*d + r)
GroupRings.fmac!(result, view(Q,:,idx), view(Q,:,idx), pm)
end
return result
end
function compute_SOS(RG::GroupRing, Q::AbstractArray)
function compute_SOS(RG::GroupRing, Q::AbstractMatrix)
result = compute_SOS(RG.pm, Q)
return GroupRingElem(result, RG)
end
function augIdproj(Q::AbstractMatrix{T}) where {T<:Real}
function augIdproj(Q::AbstractMatrix{<:Real})
result = zeros(size(Q))
l = size(Q, 2)
Threads.@threads for j in 1:l