From 33fe4bebfeeec7d17e426fce501dcfa1dc138959 Mon Sep 17 00:00:00 2001 From: liulixinkerry Date: Sun, 28 Apr 2024 22:01:51 +0800 Subject: [PATCH] fix bug when running ispd05/15 --- src/run_placement_nesterov.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/run_placement_nesterov.py b/src/run_placement_nesterov.py index 0ef60c9..cc4b541 100644 --- a/src/run_placement_nesterov.py +++ b/src/run_placement_nesterov.py @@ -12,8 +12,8 @@ def get_trunc_node_pos_fn(mov_node_size, data): def run_placement_main_nesterov(args, logger): total_start = time.time() - data, rawdb, gpdb = load_dataset(args, logger) setup_dataset_args(args) + data, rawdb, gpdb = load_dataset(args, logger) device = torch.device( "cuda:{}".format(args.gpu) if torch.cuda.is_available() else "cpu" )