mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
e7b2430818
add `class FPGMStructured` add `function FPGM_structured()` add `function _validate_distance_type()` add `function _compute_distance()` Implement method mentioned in issue #39765 --- FPGMSparsifier Implement with the new pytorch pruning API torch.ao.pruning. It is a structured pruning method, and it is added under torch.ao.pruning._experimental. Test cases are added at `test_structured_sparsifier.py`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/95689 Approved by: https://github.com/jcaip
47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
# Owner(s): ["module: unknown"]
|
|
|
|
from torch.testing._internal.common_utils import run_tests, IS_ARM64
|
|
|
|
# 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
|
|
|
|
# Structured Pruning
|
|
from ao.sparsity.test_structured_sparsifier import TestBaseStructuredSparsifier # noqa: F401
|
|
from ao.sparsity.test_structured_sparsifier import TestSaliencyPruner # noqa: F401
|
|
from ao.sparsity.test_structured_sparsifier import TestFPGMPruner # noqa: F401
|
|
|
|
# Scheduler
|
|
from ao.sparsity.test_scheduler import TestScheduler # noqa: F401
|
|
from ao.sparsity.test_scheduler import TestCubicScheduler # noqa: F401
|
|
|
|
# Composability
|
|
if not IS_ARM64:
|
|
from ao.sparsity.test_composability import TestComposability # noqa: F401
|
|
from ao.sparsity.test_composability import TestFxComposability # 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
|
|
from ao.sparsity.test_data_sparsifier import TestQuantizationUtils # noqa: F401
|
|
|
|
# Data Scheduler
|
|
from ao.sparsity.test_data_scheduler import TestBaseDataScheduler # noqa: F401
|
|
|
|
# Activation Sparsifier
|
|
from ao.sparsity.test_activation_sparsifier import TestActivationSparsifier # noqa: F401
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|