Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
Sage Moore 2025-09-26 18:27:30 +00:00
parent 1fc9de4c1a
commit 2f29120a09

View File

@ -340,7 +340,6 @@ class UBatchWrapper:
if cudagraph_runtime_mode is CUDAGraphMode.FULL:
assert batch_descriptor is not None
if batch_descriptor.num_tokens in self.cudagraphs:
logger.info("DISABLEING CUDAGRAPHS")
cudagraph_runtime_mode = CUDAGraphMode.NONE
if cudagraph_runtime_mode in (CUDAGraphMode.NONE,