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

[autoparallel] Add conv handler to generate strategies and costs info for conv (#1467)

Browse Source
This commit is contained in:
YuliangLiu0306
2022-08-19 14:57:23 +08:00
committed by GitHub
parent 1b491ad7de
commit 26a37b5cd5
4 changed files with 554 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/tensor/sharding_spec.py
View File

@@ -199,7 +199,7 @@ class ShardingSpec:
if not dim_spec.is_replica:
if index not in new_dim_partition_dict:
new_dim_partition_dict[index] = []
new_dim_partition_dict[index].append(dim_spec.shard_list)
new_dim_partition_dict[index].extend(dim_spec.shard_list)
self.dim_partition_dict = new_dim_partition_dict
def sharding_sequence_difference(self, other):