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
builder.py	add pytorch hooks (#179)	2022-01-25 22:20:54 +08:00
pipeline.py	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00