Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "93f62dd15288379355d5f4f41c3959e681f5ac41"
Commit c695369a authored by Zangwei's avatar Zangwei Committed by Frank Lee
Browse files

fix format constants.py (#358)

parent 4a0f8c2c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/constants.py
View file @ c695369a
...@@ -24,7 +24,7 @@ INPUT_GROUP_3D = 'input_group_3d' ...@@ -24,7 +24,7 @@ INPUT_GROUP_3D = 'input_group_3d'
WEIGHT_GROUP_3D = 'weight_group_3d' WEIGHT_GROUP_3D = 'weight_group_3d'
OUTPUT_GROUP_3D = 'output_group_3d' OUTPUT_GROUP_3D = 'output_group_3d'
# Attributes of tensor parallel parameters # Attributes of tensor parallel parameters
IS_TENSOR_PARALLEL = 'is_tensor_parallel' IS_TENSOR_PARALLEL = 'is_tensor_parallel'
NUM_PARTITIONS = 'num_partitions' NUM_PARTITIONS = 'num_partitions'
TENSOR_PARALLEL_ATTRIBUTES = [IS_TENSOR_PARALLEL, NUM_PARTITIONS] TENSOR_PARALLEL_ATTRIBUTES = [IS_TENSOR_PARALLEL, NUM_PARTITIONS]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment