mirror of https://github.com/hpcaitech/ColossalAI
2dd01e3a14
* [gemini] fix param op hook when output is tuple * [gemini] fix param op hook |
||
---|---|---|
.. | ||
d_tensor | ||
moe_tensor | ||
__init__.py | ||
colo_parameter.py | ||
colo_tensor.py | ||
comm_spec.py | ||
param_op_hook.py | ||
shape_consistency.py | ||
sharding_spec.py | ||
utils.py |