Skip to content

Commit 3c1b63d

Browse files
committed
modify input file path
1 parent 7a8567f commit 3c1b63d

File tree

4 files changed

+7
-6
lines changed

4 files changed

+7
-6
lines changed

graph_net/config/get_fusible_subgraph_sample_list.txt renamed to graph_net/test/dev_model_list/get_fusible_subgraph_sample_list.txt

File renamed without changes.

graph_net/config/small_sample_list_for_get_fusible_subgraph.txt renamed to graph_net/test/dev_model_list/small_sample_list_for_get_fusible_subgraph.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
#samples/timm/crossvit_small_240.in1k
22
#samples/timm/poolformerv2_s12.sail_in1k
3-
samples/timm/regnety_080.pycls_in1k
3+
#samples/timm/regnety_080.pycls_in1k
44
#samples/timm/dla46x_c.in1k
55
#samples/timm/mobilenetv1_100.ra4_e3600_r224_in1k
6-
#samples/timm/efficientnetv2_rw_s.ra2_in1k
6+
samples/timm/efficientnetv2_rw_s.ra2_in1k
77
#samples/timm/vit_base_patch16_rope_ape_224.naver_in1k
88
#samples/timm/fastvit_t8.apple_dist_in1k
99
#samples/timm/test_byobnet.r160_in1k

graph_net/test/fully_fusible_subgraph_extractor_test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ os.path.dirname(graph_net.__file__))")
66
# input model path
77
MODEL_NAME=resnet18
88
MODEL_PATH_IN_SAMPLES=/timm/$MODEL_NAME
9-
# INPUT_MODEL_LIST=$GRAPH_NET_ROOT/config/get_fusible_subgraph_sample_list.txt
10-
INPUT_MODEL_LIST=$GRAPH_NET_ROOT/config/small_sample_list_for_get_fusible_subgraph.txt
9+
# INPUT_MODEL_LIST=$GRAPH_NET_ROOT/test/dev_model_list/get_fusible_subgraph_sample_list.txt
10+
INPUT_MODEL_LIST=$GRAPH_NET_ROOT/test/dev_model_list/small_sample_list_for_get_fusible_subgraph.txt
1111

1212
OUTPUT_DIR="/tmp/find_fully_fusible_output"
1313
config_json_str=$(cat <<EOF

graph_net/torch/graph_decomposer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,15 +130,16 @@ def __call__(self, rel_model_path):
130130
module, inputs = get_torch_module_and_inputs(model_path)
131131
gm = parse_immutable_model_path_into_sole_graph_module(model_path)
132132
try:
133-
logger.warning("convert_to_submodules_graph-call-begin")
133+
# logger.warning("convert_to_submodules_graph-call-begin")
134134
rewrited_gm: torch.fx.GraphModule = convert_to_submodules_graph(
135135
gm,
136136
submodule_hook=self.get_naive_decomposer_extractor(model_path),
137137
**config,
138138
)
139139
rewrited_gm(*inputs)
140140
finally:
141-
logger.warning("convert_to_submodules_graph-call-end")
141+
pass
142+
# logger.warning("convert_to_submodules_graph-call-end")
142143

143144
def get_naive_decomposer_extractor(self, model_path):
144145
def fn(submodule, seq_no):

0 commit comments

Comments
 (0)