github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-12 20:54:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)

Browse Source
This commit is contained in:
digger yu
2023-05-24 09:01:50 +08:00
committed by GitHub
parent 725365f297
commit 7f8203af69
19 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/fx/passes/passes_for_gpt2_test.py
View File

@@ -230,7 +230,7 @@ def split_module_for_gpt2_test(
use_partition.partitions_dependent_on.setdefault(def_partition_name)
node_process_list = list(m.graph.nodes)
# split nodes into parititons
# split nodes into partitions
while node_process_list:
node = node_process_list.pop(0)
orig_nodes[node.name] = node
@@ -277,7 +277,7 @@ def split_module_for_gpt2_test(
if len(sorted_partitions) != len(partitions):
raise RuntimeError("cycle exists between partitions!")
# add placeholders to parititons
# add placeholders to partitions
for partition_name in sorted_partitions:
partition = partitions[partition_name]
for input in partition.inputs: