| 1912 | } |
| 1913 | |
| 1914 | void HeartMuLaWeightsRuntime::reset_backbone_cached_state( |
| 1915 | HeartMuLaBackboneCachedState & state, |
| 1916 | runtime::TransformerKVState prefill_state) const { |
| 1917 | state.prefill_steps_ = prefill_state.current_end; |
| 1918 | state.pending_state_ = std::move(prefill_state); |
| 1919 | state.graph_has_state_ = false; |
| 1920 | } |
| 1921 | |
| 1922 | HeartMuLaBackboneResult HeartMuLaWeightsRuntime::backbone_cached_step( |
| 1923 | const std::vector<float> & embedding, |
no outgoing calls
no test coverage detected