lxchen/Xplace_for_ICCAD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug when running ispd05/15

Browse Source
This commit is contained in:
liulixinkerry 2024-04-28 22:01:51 +08:00
parent dfbc668017
commit 33fe4bebfe
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/run_placement_nesterov.py
View File

@ -12,8 +12,8 @@ def get_trunc_node_pos_fn(mov_node_size, data):
def run_placement_main_nesterov(args, logger): def run_placement_main_nesterov(args, logger):
total_start = time.time() total_start = time.time()
data, rawdb, gpdb = load_dataset(args, logger)
setup_dataset_args(args) setup_dataset_args(args)
data, rawdb, gpdb = load_dataset(args, logger)
device = torch.device( device = torch.device(
"cuda:{}".format(args.gpu) if torch.cuda.is_available() else "cpu" "cuda:{}".format(args.gpu) if torch.cuda.is_available() else "cpu"
) )