mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
fa6b6842e1
Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #79774 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79774 Approved by: https://github.com/z-a-f
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Owner(s): ["module: unknown"]
|
|
|
|
from torch.testing._internal.common_utils import run_tests
|
|
|
|
# Kernels
|
|
from ao.sparsity.test_kernels import TestQuantizedSparseKernels # noqa: F401
|
|
from ao.sparsity.test_kernels import TestQuantizedSparseLayers # noqa: F401
|
|
|
|
# Parametrizations
|
|
from ao.sparsity.test_parametrization import TestFakeSparsity # noqa: F401
|
|
|
|
# Sparsifier
|
|
from ao.sparsity.test_sparsifier import TestBaseSparsifier # noqa: F401
|
|
from ao.sparsity.test_sparsifier import TestWeightNormSparsifier # noqa: F401
|
|
from ao.sparsity.test_sparsifier import TestNearlyDiagonalSparsifier # noqa: F401
|
|
|
|
# Pruner
|
|
from ao.sparsity.test_pruner import TestBasePruner # noqa: F401
|
|
|
|
# Scheduler
|
|
from ao.sparsity.test_scheduler import TestScheduler # noqa: F401
|
|
|
|
# Composability
|
|
from ao.sparsity.test_composability import TestComposability # noqa: F401
|
|
|
|
# Utilities
|
|
from ao.sparsity.test_sparsity_utils import TestSparsityUtilFunctions # noqa: F401
|
|
|
|
# Data Sparsifier
|
|
from ao.sparsity.test_data_sparsifier import TestBaseDataSparsifier # noqa: F401
|
|
from ao.sparsity.test_data_sparsifier import TestNormDataSparsifiers # noqa: F401
|
|
|
|
# Data Scheduler
|
|
from ao.sparsity.test_data_scheduler import TestBaseDataScheduler # noqa: F401
|
|
|
|
if __name__ == '__main__':
|
|
run_tests()
|