Skip to content

Commit

Permalink
update CDC benchmark
Browse files Browse the repository at this point in the history
  • Loading branch information
frapac committed Aug 20, 2024
1 parent 5a62278 commit 304386c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
4 changes: 1 addition & 3 deletions HybridKKT.jl/benchmarks/cdc/benchmark.jl
Original file line number Diff line number Diff line change
Expand Up @@ -79,8 +79,6 @@ end
flag = quick ? "short" : "full"
print_level = verbose ? MadNLP.INFO : MadNLP.ERROR

# if quick

if solver == "all" || solver == "ma27"
@info "[CPU] Benchmark SparseKKTSystem+ma27"
results = run_benchmark(
Expand Down Expand Up @@ -162,7 +160,7 @@ end
use_gpu=true,
tol=tol,
linear_solver=MadNLPGPU.CUDSSSolver,
cudss_algorithm=MadNLP.BUNCHKAUFMAN,
cudss_algorithm=MadNLP.LDL,
print_level=print_level,
)
output_file = joinpath(RESULTS_DIR, "cops-$(flag)-madnlp-hckkt-cudss-ldl.csv")
Expand Down
2 changes: 1 addition & 1 deletion HybridKKT.jl/benchmarks/cdc/distillation.jl
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ function distillation_column_model(N = 3; T = Float64, backend = nothing, kwargs
itr1 = ExaModels.convert_array(collect(Iterators.product(1:N, FT+1:NT)), backend)
itr2 = ExaModels.convert_array(collect(Iterators.product(0:N, 0:NT+1)), backend)

c = ExaModels.ExaCore(T, backend)
c = ExaModels.ExaCore(T; backend=backend)

xA = ExaModels.variable(c, 0:N, 0:NT+1; start = 0.5)
yA = ExaModels.variable(c, 0:N, 0:NT+1; start = 0.5)
Expand Down

0 comments on commit 304386c

Please sign in to comment.