github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_autochunk/test_autochunk_vit
History
Xuanlei Zhao 10c61de2f7
[autochunk] support vit (#3084) 
support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit
 		2023-03-10 10:23:26 +08:00
..
test_autochunk_vit.py [autochunk] support vit (#3084) 2023-03-10 10:23:26 +08:00
test_autochunk_vit_utils.py [autochunk] support vit (#3084) 2023-03-10 10:23:26 +08:00