Skip to content

Commit 64ddd11

Browse files
committed
fix
1 parent 4fe2cd0 commit 64ddd11

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graph_net/tools/generate_subgraph_dataset.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ function rename_fusible_subgraph() {
218218
echo ">>> [8] Rename subgraph samples under ${FUSIBLE_SUBGRAPH_SAMPLES_DIR}."
219219
echo ">>>"
220220
python3 -m graph_net.model_path_handler \
221-
--model-path-list ${range_decomposed_subgraph_list} \
221+
--model-path-list ${fusible_subgraph_list} \
222222
--handler-config=$(base64 -w 0 <<EOF
223223
{
224224
"handler_path": "$GRAPH_NET_ROOT/graph_net/torch/graph_variable_renamer.py",
@@ -257,7 +257,7 @@ function generate_unittests() {
257257
"handler_class_name": "AgentUnittestGeneratorPass",
258258
"handler_config": {
259259
"framework": "torch",
260-
"model_path_prefix": "${DEVICE_REWRITED_OUTPUT_DIR}",
260+
"model_path_prefix": "${DEDUPLICATED_FUSIBLE_SUBGRAPH_DIR}",
261261
"output_dir": "$UNITTESTS_OUTPUT_DIR",
262262
"device": "cuda",
263263
"generate_main": false,

0 commit comments

Comments
 (0)