mirror of https://github.com/hpcaitech/ColossalAI
c2fdc6a011
* derive compute pattern from dist spec * polish code |
||
---|---|---|
.. | ||
__init__.py | ||
addmm.py | ||
element_wise.py | ||
embedding.py | ||
layernorm.py | ||
linear.py | ||
loss.py |