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