github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/engine/ophooks
History
Jiarui Fang 569357fea0
add pytorch hooks (#179) 
* add pytorch hooks
fix #175

* remove licenses in src code

* add gpu memory tracer

* replacing print with logger in ophooks.
 		2022-01-25 22:20:54 +08:00
..
__init__.py add pytorch hooks (#179) 2022-01-25 22:20:54 +08:00
_base_ophook.py add pytorch hooks (#179) 2022-01-25 22:20:54 +08:00
_memtracer_ophook.py add pytorch hooks (#179) 2022-01-25 22:20:54 +08:00