[NPU]remove duplicate code (#31863)
This commit is contained in:
@@ -323,8 +323,6 @@ class AscendGDNAttnBackend(AscendMambaAttnBackendBase):
|
||||
ssm_states.dtype, copy=False
|
||||
)
|
||||
ssm_states[cache_indices] = last_recurrent_state
|
||||
last_recurrent_state = last_recurrent_state.to(ssm_states.dtype, copy=False)
|
||||
ssm_states[cache_indices] = last_recurrent_state
|
||||
if h is not None:
|
||||
self._track_mamba_state_extend(
|
||||
forward_batch, h, ssm_states, forward_metadata
|
||||
|
||||
Reference in New Issue
Block a user