[v1][torch.compile] support managing cudagraph buffer (#10203)

Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
youkaichao
2024-11-11 11:10:27 -08:00
committed by GitHub
parent d7a4f2207b
commit 330e82d34a
4 changed files with 59 additions and 8 deletions

View File

@@ -1,4 +1,5 @@
{
"use_cudagraph": true,
"non_cudagraph_ops": ["silly.attention"]
"non_cudagraph_ops": ["silly.attention"],
"cudagraph_copy_inputs": true
}