mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
d176c82bd5
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66411 The original tests were disabled, and had some bugs. This fixes those unittests. Test Plan: Imported from OSS Reviewed By: HDCharles Differential Revision: D31590678 Pulled By: z-a-f fbshipit-source-id: ddbed34cc01d5f15580cb8f0033416f2f9780068
25 lines
743 B
Python
25 lines
743 B
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
|
|
|
|
# Pruner
|
|
from ao.sparsity.test_pruner import TestBasePruner # noqa: F401
|
|
|
|
# Scheduler
|
|
from ao.sparsity.test_scheduler import TestScheduler # noqa: F401
|
|
|
|
if __name__ == '__main__':
|
|
run_tests()
|