minor change

JewelRoam · JewelRoam · commit 8a2a9cd9a353 · 2025-12-23T20:08:22.000+08:00
diff --git a/graph_net/tools/generate_subgraph_dataset.sh b/graph_net/tools/generate_subgraph_dataset.sh
@@ -163,6 +163,7 @@ function cumsum_num_kernels() {
         "output_json_file_name": "cumsum_num_kernels.json",
         "model_path_prefix": "$2",
         "output_dir": "$3",
+        "device": "cuda",
         "resume": ${RESUME}
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -163,6 +163,7 @@ function cumsum_num_kernels() {`
`163`	`163`	`"output_json_file_name": "cumsum_num_kernels.json",`
`164`	`164`	`"model_path_prefix": "$2",`
`165`	`165`	`"output_dir": "$3",`
	`166`	`+ "device": "cuda",`
`166`	`167`	`"resume": ${RESUME}`
`167`	`168`	`}`
`168`	`169`	`}`