ver217 6ae8ed0407 [lazyinit] add correctness verification (#3147) ... * [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		2023-03-17 13:49:04 +08:00
..
checkpoint	…
checkpoint_io	…
data_sampler	…
model	[lazyinit] add correctness verification (#3147)	2023-03-17 13:49:04 +08:00
multi_tensor_apply	[setup] support pre-build and jit-build of cuda kernels (#2374)	2023-01-06 20:50:26 +08:00
profiler	…
rank_recorder	…
tensor_detector	…
__init__.py	[ddp] add is_ddp_ignored (#2434)	2023-01-11 12:22:45 +08:00
activation_checkpoint.py	…
checkpointing.py	…
common.py	Fix port exception type (#2925)	2023-02-28 11:00:43 +08:00
cuda.py	…
memory.py	…
moe.py	…
timer.py	…