Super tiny fix main code (#16432)

This commit is contained in:
fzyzcjy
2026-01-05 10:34:25 +08:00
committed by GitHub
parent e267ca0beb
commit 1e7b326482

View File

@@ -534,7 +534,7 @@ class CudaGraphRunner:
)
graph_fn = (
partial(memory_saver_adapter.cuda_graph, tag=GPU_MEMORY_TYPE_CUDA_GRAPH)
if memory_saver_adapter.log_timing
if memory_saver_adapter.enabled
else self.device_module.graph
)
with graph_fn(cuda_graph=graph, pool=pool, stream=stream):