| 2095 | } |
| 2096 | |
| 2097 | void HeartMuLaWeightsRuntime::release_graph_workspaces() const { |
| 2098 | if (backbone_prefill_graph_ != nullptr) { |
| 2099 | backbone_prefill_graph_->release_workspace(); |
| 2100 | } |
| 2101 | if (backbone_cached_step_graph_ != nullptr) { |
| 2102 | backbone_cached_step_graph_->release_workspace(); |
| 2103 | } |
| 2104 | if (decoder_prefill_graph_ != nullptr) { |
| 2105 | decoder_prefill_graph_->release_workspace(); |
| 2106 | } |
| 2107 | if (decoder_cached_step_graph_ != nullptr) { |
| 2108 | decoder_cached_step_graph_->release_workspace(); |
| 2109 | } |
| 2110 | if (prompt_frame_embedding_graph_ != nullptr) { |
| 2111 | prompt_frame_embedding_graph_->release_workspace(); |
| 2112 | } |
| 2113 | if (step_frame_embedding_graph_ != nullptr) { |
| 2114 | step_frame_embedding_graph_->release_workspace(); |
| 2115 | } |
| 2116 | } |
| 2117 | |
| 2118 | void HeartMuLaWeightsRuntime::clear_graph_cache() const { |
| 2119 | backbone_prefill_graph_.reset(); |