mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
26f4f10ac8
The `usort` config in `pyproject.toml` has no effect due to a typo. Fixing the typo make `usort` do more and generate the changes in the PR. Except `pyproject.toml`, all changes are generated by `lintrunner -a --take UFMT --all-files`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127126 Approved by: https://github.com/kit1980
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
# Owner(s): ["module: unknown"]
|
|
|
|
# Kernels
|
|
from ao.sparsity.test_kernels import ( # noqa: F401 # noqa: F401
|
|
TestQuantizedSparseKernels,
|
|
TestQuantizedSparseLayers,
|
|
)
|
|
|
|
# Parametrizations
|
|
from ao.sparsity.test_parametrization import TestFakeSparsity # noqa: F401
|
|
|
|
# Scheduler
|
|
from ao.sparsity.test_scheduler import ( # noqa: F401 # noqa: F401
|
|
TestCubicScheduler,
|
|
TestScheduler,
|
|
)
|
|
|
|
# Sparsifier
|
|
from ao.sparsity.test_sparsifier import ( # noqa: F401 # noqa: F401 # noqa: F401
|
|
TestBaseSparsifier,
|
|
TestNearlyDiagonalSparsifier,
|
|
TestWeightNormSparsifier,
|
|
)
|
|
|
|
# Structured Pruning
|
|
from ao.sparsity.test_structured_sparsifier import ( # noqa: F401 # noqa: F401 # noqa: F401
|
|
TestBaseStructuredSparsifier,
|
|
TestFPGMPruner,
|
|
TestSaliencyPruner,
|
|
)
|
|
|
|
from torch.testing._internal.common_utils import IS_ARM64, run_tests
|
|
|
|
# Composability
|
|
if not IS_ARM64:
|
|
from ao.sparsity.test_composability import ( # noqa: F401 # noqa: F401
|
|
TestComposability,
|
|
TestFxComposability,
|
|
)
|
|
|
|
# Activation Sparsifier
|
|
from ao.sparsity.test_activation_sparsifier import ( # noqa: F401
|
|
TestActivationSparsifier,
|
|
)
|
|
|
|
# Data Scheduler
|
|
from ao.sparsity.test_data_scheduler import TestBaseDataScheduler # noqa: F401
|
|
|
|
# Data Sparsifier
|
|
from ao.sparsity.test_data_sparsifier import ( # noqa: F401 # noqa: F401 # noqa: F401
|
|
TestBaseDataSparsifier,
|
|
TestNormDataSparsifiers,
|
|
TestQuantizationUtils,
|
|
)
|
|
|
|
# Utilities
|
|
from ao.sparsity.test_sparsity_utils import TestSparsityUtilFunctions # noqa: F401
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|