diff --git a/main.jl b/main.jl index c84bccd..d99e5bf 100644 --- a/main.jl +++ b/main.jl @@ -1,8 +1,21 @@ +using AMDGPU using Milepost7 using MPI +using Logging + MPI.Init() +if MPI.Comm_rank(MPI.COMM_WORLD) != 0 + disable_logging(Info) + disable_logging(Warn) +else + if AMDGPU.has_rocm_gpu() + @show AMDGPU.libhsaruntime_path + @show AMDGPU.use_artifacts + end +end + if !isinteractive() args = parse_cmd() profile = args["profile"] @@ -41,4 +54,4 @@ info = milepost7( proxal_iter=proxal_iter, exatron_inner_iter=exatron_inner_iter, exatron_outer_iter=exatron_outer_iter -); \ No newline at end of file +); diff --git a/src/Milepost7.jl b/src/Milepost7.jl index 3bbebf2..eddc8fe 100644 --- a/src/Milepost7.jl +++ b/src/Milepost7.jl @@ -18,20 +18,6 @@ using ProxAL export milepost7, parse_cmd -function ProxAL.ExaAdmm.KAArray{T}(n::Int, device::CUDABackend) where {T} - return CuArray{T}(undef, n) -end -function ProxAL.ExaAdmm.KAArray{T}(n1::Int, n2::Int, device::CUDABackend) where {T} - return CuArray{T}(undef, n1, n2) -end - -function ProxAL.ExaAdmm.KAArray{T}(n::Int, device::ROCBackend) where {T} - return ROCArray{T}(undef, n) -end -function ProxAL.ExaAdmm.KAArray{T}(n1::Int, n2::Int, device::ROCBackend) where {T} - return ROCArray{T}(undef, n1, n2) -end - function milepost7( case::String, demandfiles::String, @@ -74,6 +60,7 @@ algparams.verbose = 1 algparams.tol = 1e-3 algparams.decompCtgs = (K > 0) algparams.iterlim = proxal_iter +algparams.verbose_inner = 0 # choose backend if configuration == 1 diff --git a/test/runtests.jl b/test/runtests.jl index ed87539..ada9879 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -40,14 +40,16 @@ MPI.Init() maxviol_c_actual = info.maxviol_c_actual[end] maxviol_d = info.maxviol_d[end] end - @testset "Testing on CUDA" begin - info = milepost7(case, load, T, K, 3; profile=false, rhopq=rhopq, rhova=rhova, proxal_iter=proxal_iter) + if CUDA.has_cuda_gpu() + @testset "Testing on CUDA" begin + info = milepost7(case, load, T, K, 3; profile=false, rhopq=rhopq, rhova=rhova, proxal_iter=proxal_iter) - @test isapprox(info.maxviol_t[end], maxviol_t) - @test isapprox(info.maxviol_t_actual[end], maxviol_t_actual) - @test isapprox(info.maxviol_c[end], maxviol_c) - @test isapprox(info.maxviol_c_actual[end], maxviol_c_actual) - @test isapprox(info.maxviol_d[end], maxviol_d) + @test isapprox(info.maxviol_t[end], maxviol_t) + @test isapprox(info.maxviol_t_actual[end], maxviol_t_actual) + @test isapprox(info.maxviol_c[end], maxviol_c) + @test isapprox(info.maxviol_c_actual[end], maxviol_c_actual) + @test isapprox(info.maxviol_d[end], maxviol_d) + end end @testset "Testing on KA" begin info = milepost7(case, load, T, K, 4; profile=false, rhopq=rhopq, rhova=rhova, proxal_iter=proxal_iter)