mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
YuliangLiu0306
36c0f3ea5b
|
2 years ago | |
---|---|---|
.. | ||
__init__.py | 2 years ago | |
colo_parameter.py | 2 years ago | |
colo_tensor.py | 2 years ago | |
comm_spec.py | 2 years ago | |
compute_spec.py | ||
const.py | ||
dist_spec_mgr.py | 2 years ago | |
distspec.py | ||
op_wrapper.py | ||
param_op_hook.py | 2 years ago | |
process_group.py | 2 years ago | |
shape_consistency.py | 2 years ago | |
sharding_spec.py | 2 years ago | |
tensor_spec.py | 2 years ago | |
utils.py | 2 years ago |