github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 12:30:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

[lazyinit] add correctness verification (#3147)

Browse Source 
* [lazyinit] fix shared module

* [tests] add lazy init test utils

* [tests] add torchvision for lazy init

* [lazyinit] fix pre op fn

* [lazyinit] handle legacy constructor

* [tests] refactor lazy init test models

* [tests] refactor lazy init test utils

* [lazyinit] fix ops don't support meta

* [tests] lazy init test timm models

* [lazyinit] fix set data

* [lazyinit] handle apex layers

* [tests] lazy init test transformers models

* [tests] lazy init test torchaudio models

* [lazyinit] fix import path

* [tests] lazy init test torchrec models

* [tests] update torch version in CI

* [tests] revert torch version in CI

* [tests] skip lazy init test
This commit is contained in:
ver217
2023-03-17 13:49:04 +08:00
committed by GitHub
parent 3c01280a56
commit 6ae8ed0407
4 changed files with 226 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/kit/model_zoo/torchrec/__init__.py Normal file
View File

@@ -0,0 +1 @@
from .torchrec import *