github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 07:28:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
7be397ca9c3cb67bff614e4bd44e836d22a693d0
ColossalAI/colossalai/builder
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
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