Skip to content

Commit d42524d

Browse files
committed
corrrect 'fusable' -> 'fusible'
1 parent df1ea34 commit d42524d

File tree

4 files changed

+11
-11
lines changed

4 files changed

+11
-11
lines changed

graph_net/test/graph_decompose_and_look_for_fully_fusable_subgraph_test.sh renamed to graph_net/test/graph_decompose_and_look_for_fully_fusible_subgraph_test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,9 @@ decorator_config_json_str=$(cat <<EOF
1111
"decorator_path": "$GRAPH_NET_ROOT/torch/extractor.py",
1212
"decorator_config": {
1313
"name": "$MODEL_NAME",
14-
"custom_extractor_path": "$GRAPH_NET_ROOT/torch/fully_fusable_subgraph_extractor.py",
14+
"custom_extractor_path": "$GRAPH_NET_ROOT/torch/fully_fusible_subgraph_extractor.py",
1515
"custom_extractor_config": {
16-
"output_dir": "/tmp/find_fully_fusable_output",
16+
"output_dir": "/tmp/find_fully_fusible_output",
1717
"split_positions": [],
1818
"group_head_and_tail": true,
1919
"max_step": 3,

graph_net/test/naive_decomposer_and_post_extract_process_test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ decorator_config_json_str=$(cat <<EOF
2020
"filter_path":"$GRAPH_NET_ROOT/torch/naive_subgraph_filter.py",
2121
"filter_config": {},
2222
"post_extract_process_path":"$GRAPH_NET_ROOT/torch/post_extract_process_count_kernels.py",
23-
"post_extract_process_class_name": "GraphFullyFusable"
23+
"post_extract_process_class_name": "GraphFullyFusible"
2424
}
2525
}
2626
}

graph_net/torch/fully_fusable_subgraph_extractor.py renamed to graph_net/torch/fully_fusible_subgraph_extractor.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def _build_decompose_config(
8282
self.config["split_positions"] = [start_pos, end_pos]
8383
graph_net_root = os.path.dirname(graph_net.__file__)
8484

85-
check_fusable_config = {
85+
check_fusible_config = {
8686
"decorator_path": f"{graph_net_root}/torch/extractor.py",
8787
"decorator_config": {
8888
"name": f"{self.name}",
@@ -94,28 +94,28 @@ def _build_decompose_config(
9494
"filter_path": f"{graph_net_root}/torch/naive_subgraph_filter.py",
9595
"filter_config": {},
9696
"post_extract_process_path": f"{graph_net_root}/torch/post_extract_process_count_kernels.py",
97-
"post_extract_process_class_name": "GraphFullyFusable",
97+
"post_extract_process_class_name": "GraphFullyFusible",
9898
},
9999
},
100100
}
101-
return check_fusable_config
101+
return check_fusible_config
102102

103103
def __call__(self, gm: torch.fx.GraphModule, sample_inputs):
104104
for start_pos, end_pos in self._get_sub_ranges():
105105
with tempfile.TemporaryDirectory(
106-
prefix="_find_fusable_subgraph_"
106+
prefix="_find_fusible_subgraph_"
107107
) as temp_dir:
108-
check_fusable_config = self._build_decompose_config(
108+
check_fusible_config = self._build_decompose_config(
109109
temp_dir, start_pos, end_pos
110110
)
111111
print("current split_positions:", self.config["split_positions"])
112-
success = constraint_util.RunModelPredicator(check_fusable_config)(
112+
success = constraint_util.RunModelPredicator(check_fusible_config)(
113113
self.config["model_path"]
114114
)
115115
if success:
116116
target_path = self._handle_success(temp_dir, start_pos, end_pos)
117117
print(
118-
f"SUCCESS in finding the biggest fully fusable subgraph. Result saved to: {target_path}"
118+
f"SUCCESS in finding the biggest fully fusible subgraph. Result saved to: {target_path}"
119119
)
120120
break
121121
return gm.forward

graph_net/torch/post_extract_process_count_kernels.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from torch.profiler import profile, record_function, ProfilerActivity
77

88

9-
class GraphFullyFusable:
9+
class GraphFullyFusible:
1010
def __init__(self, config):
1111
self.config = config
1212

0 commit comments

Comments
 (0)