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.
YH
2629f9717d
|
2 years ago | |
---|---|---|
.. | ||
d_tensor | 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 | 2 years ago | |
const.py | ||
dist_spec_mgr.py | 2 years ago | |
distspec.py | 2 years ago | |
op_wrapper.py | 2 years ago | |
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 |