From ca749f3d541bee52bae4b0d42ad885a88754755c Mon Sep 17 00:00:00 2001 From: Kai Xu Date: Thu, 10 Sep 2026 19:46:01 -0700 Subject: [PATCH] Sort DASC modules by explicit name Signed-off-by: Kai Xu --- modelopt/torch/sparsity/state_sparsity/policy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelopt/torch/sparsity/state_sparsity/policy.py b/modelopt/torch/sparsity/state_sparsity/policy.py index e7d84351b57..098dbcb4a59 100644 --- a/modelopt/torch/sparsity/state_sparsity/policy.py +++ b/modelopt/torch/sparsity/state_sparsity/policy.py @@ -170,7 +170,7 @@ def _get_gdn_modules(model: nn.Module) -> dict[str, nn.Module]: f"{module_name}.{class_name}" for module_name, class_name in _SUPPORTED_GDN_CLASS_PATHS ) raise ApplyModeError(f"DASC found no supported GDN modules; expected one of: {supported}") - return dict(sorted(identity_modules)) + return dict(sorted(identity_modules, key=lambda item: item[0])) def _analyze_gdn_modules(