Skip to content

Commit 1b5d448

Browse files
Add Deprecation warning for GradNAS
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
1 parent acabf87 commit 1b5d448

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

modelopt/torch/prune/gradnas.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,10 @@ def default_search_config(self) -> SearchConfig:
181181

182182
def before_search(self) -> None:
183183
"""Setup search with gradient-based score."""
184+
warnings.warn(
185+
"GradNAS pruning algorithm is deprecated and will be removed in the future.",
186+
DeprecationWarning,
187+
)
184188
# initialize here to keep DMRegistry empty during imports
185189
GradientBinarySearcher.SETUP_GRADIENT_FUNC = {
186190
DMRegistry[nn.Linear]: _setup_grad_manager_linear,

0 commit comments

Comments
 (0)