From b00c27fd6dbd1217e4778d27da055efa999efafc Mon Sep 17 00:00:00 2001 From: yanglong_unimelb Date: Fri, 30 Jan 2026 12:13:42 +0800 Subject: [PATCH] [bugfix] hash router with dsa --- mindformers/pynative/base_models/gpt/gpt_layer_specs.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindformers/pynative/base_models/gpt/gpt_layer_specs.py b/mindformers/pynative/base_models/gpt/gpt_layer_specs.py index 88d6a594a..ca8d24e46 100644 --- a/mindformers/pynative/base_models/gpt/gpt_layer_specs.py +++ b/mindformers/pynative/base_models/gpt/gpt_layer_specs.py @@ -241,6 +241,8 @@ def get_gpt_decoder_block_spec( qk_layernorm=config.qk_layernorm, multi_latent_attention=config.multi_latent_attention, fused_norm=config.fused_norm, + experimental_attention_variant=getattr(config, 'experimental_attention_variant', None), + dsa_use_fused_ops=getattr(config, 'dsa_use_fused_ops', False), hash_routed_experts=True, ) -- Gitee