We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a31c262 commit 8f8fedeCopy full SHA for 8f8fede
1 file changed
megatron/core/inference/engines/dynamic_engine.py
@@ -725,6 +725,9 @@ def suspend(self):
725
and not self.context.static_kv_memory_pointers
726
):
727
delete_cuda_graphs()
728
+ if self.controller._sampling_backend == "flashinfer":
729
+ self.controller._sampling._sampling_cuda_graphs.clear()
730
+ self.controller._sampling._verification_cuda_graphs.clear()
731
732
# Build the list of requests to re-add on resume.
733
# All waiting requests are always included; active requests are included
0 commit comments