Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
piecewise_compilation_config.json 110 Bytes
Newer Older
youkaichao's avatar
[torch.compile] rework compile control with piecewise cudagraph (#9715)  
youkaichao committed
1
2 
{
    "use_cudagraph": true,
youkaichao's avatar
[v1][torch.compile] support managing cudagraph buffer (#10203)  
youkaichao committed
3
4 
    "non_cudagraph_ops": ["silly.attention"],
    "cudagraph_copy_inputs": true
youkaichao's avatar
[torch.compile] rework compile control with piecewise cudagraph (#9715)  
youkaichao committed
5 
}