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

[autoparallel]add essential CommActions for broadcast oprands (#1793)

Browse Source
This commit is contained in:
YuliangLiu0306
2022-11-04 18:36:42 +08:00
committed by GitHub
parent 05ce3d369f
commit e34e850a4c
9 changed files with 102 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_fx/test_tracer/test_bias_addition_module.py
View File

@@ -105,7 +105,7 @@ def test_conv_module():
assert weight_node._meta_data.shape == (6, 3, 2, 2)
assert bias_node._meta_data.shape == (6,)
assert conv_node._meta_data.shape == (4, 6, 63, 63)
assert view_node._meta_data.shape == (1, 6, 1, 1)
assert view_node._meta_data.shape == (6, 1, 1)
assert add_node._meta_data.shape == (4, 6, 63, 63)