fix bug when running ispd05/15

2024-04-28 22:01:51 +08:00 · 2024-04-28 22:01:51 +08:00 · 33fe4bebfe
commit 33fe4bebfe
parent dfbc668017
1 changed files with 1 additions and 1 deletions
--- 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"
    )