Fix fusible subgraph generation and add into graph_net/tools/generate_subgraph_dataset.sh (#480)

* Add debug case for CumSumNumKernelsGenerator

* FIx dir name

* fix name mismatch in torch/fx_graph_parse_util.py

* Add gen_fusible_subgraphs to graph_net/tools/generate_subgraph_dataset.sh

* warmup in count_kernels

---------

Co-authored-by: Xinqi Li <[email protected]>

Author: JewelRoam

graph_net/test/debug_fusible_subgraph_generate.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+GRAPH_NET_ROOT=$(python3 -c "import graph_net; import os; print(os.path.dirname(os.path.dirname(graph_net.__file__)))")
+
+model_path_list=$GRAPH_NET_ROOT/graph_net/test/workspace_cumsum_num_kernels/test_model_list.txt
+WORKSPACE_ROOT=/tmp/fusible_subgraphs
+CUMSUM_NUM_KERNELS_WORKSPACE=$WORKSPACE_ROOT/workspace_cumsum_num_kernels
+FUSIBLE_SUBGRAPH_RANGES_WORKSPACE=$WORKSPACE_ROOT/workspace_fusible_subgraph_ranges
+FUSIBLE_SUBGRAPH_SAMPLES_WORKSPACE=$WORKSPACE_ROOT/workspace_fusible_subgraph_samples
+resume=false
+
+python3 -m graph_net.model_path_handler \
+    --model-path-list "$model_path_list" \
+    --handler-config $(base64 -w 0 <<EOF
+{
+    "handler_path": "$GRAPH_NET_ROOT/graph_net/torch/sample_passes/cumsum_num_kernels_generator.py",
+    "handler_class_name": "CumSumNumKernelsGenerator",
+    "handler_config": {
+        "output_json_file_name": "cumsum_num_kernels.json",
+        "model_path_prefix": "$GRAPH_NET_ROOT/graph_net/test/workspace_cumsum_num_kernels",
+        "output_dir": "$CUMSUM_NUM_KERNELS_WORKSPACE",
+        "resume": $resume
+    }
+}
+EOF
+)

graph_net/test/workspace_cumsum_num_kernels/samples/transformers-auto-model/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification/_decomposed/hf-tiny-model-private_tiny-random-Swinv2ForImageClassification_0/graph_net.json

@@ -0,0 +1,7 @@
+{
+    "framework": "torch",
+    "num_devices_required": 1,
+    "num_nodes_required": 1,
+    "dynamic": false,
+    "model_name": "hf-tiny-model-private_tiny-random-Swinv2ForImageClassification_0"
+}