# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.

# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from torchao.quantization.quant_api import (
    int8_dynamic_activation_int8_semi_sparse_weight,
)

from .sparse_api import (
    apply_fake_sparsity,
    block_sparse_weight,
    semi_sparse_weight,
    sparsify_,
)
from .supermask import SupermaskLinear
from .utils import PerChannelNormObserver  # noqa: F403
from .wanda import WandaSparsifier  # noqa: F403

__all__ = [
    "WandaSparsifier",
    "SupermaskLinear",
    "PerChannelNormObserver",
    "apply_fake_sparsity",
    "sparsify_",
    "semi_sparse_weight",
    "block_sparse_weight",
    "int8_dynamic_activation_int8_semi_sparse_weight",
]