Super tiny fix main code (#16432)
This commit is contained in:
@@ -534,7 +534,7 @@ class CudaGraphRunner:
|
|||||||
)
|
)
|
||||||
graph_fn = (
|
graph_fn = (
|
||||||
partial(memory_saver_adapter.cuda_graph, tag=GPU_MEMORY_TYPE_CUDA_GRAPH)
|
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
|
else self.device_module.graph
|
||||||
)
|
)
|
||||||
with graph_fn(cuda_graph=graph, pool=pool, stream=stream):
|
with graph_fn(cuda_graph=graph, pool=pool, stream=stream):
|
||||||
|
|||||||
Reference in New Issue
Block a user