mirror of https://github.com/hpcaitech/ColossalAI
3dd6994427
* [fx/profiler] modify data_ptr into uuid for all tensors. * [fx] modify uuid. * [fx/profiler] tune performance on GPT-2. * [fx] updates. * [fx] debug. * [fx] debug. * [fx] cuda. |
||
---|---|---|
.. | ||
__init__.py | ||
build_c_ext.py | ||
ckpt_solver_chen.py | ||
ckpt_solver_pofo.py | ||
ckpt_solver_rotor.py | ||
dynamic_programs.c | ||
linearize.py | ||
operation.py |