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.
Frank Lee
190a6ea9c2
|
1 year ago | |
---|---|---|
.. | ||
d_tensor | 1 year ago | |
__init__.py | ||
colo_parameter.py | ||
colo_tensor.py | ||
comm_spec.py | 1 year ago | |
compute_spec.py | ||
const.py | ||
dist_spec_mgr.py | ||
distspec.py | ||
op_wrapper.py | ||
param_op_hook.py | ||
process_group.py | ||
shape_consistency.py | 1 year ago | |
sharding_spec.py | 1 year ago | |
tensor_spec.py | ||
utils.py |