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.
ver217
7faef93326
|
3 years ago | |
---|---|---|
.. | ||
_ops | 3 years ago | |
graph | ||
modules | 3 years ago | |
optim | ||
__init__.py | 3 years ago | |
chunk.py | 3 years ago | |
colo_parameter.py | 3 years ago | |
colo_tensor.py | 3 years ago | |
const.py | ||
dist_spec_mgr.py | 3 years ago | |
distspec.py | 3 years ago | |
module_utils.py | 3 years ago | |
op_wrapper.py | ||
param_op_hook.py | 3 years ago | |
spec.py | 3 years ago | |
utils.py |