torch_cudagraph_fx is a make_fx-style tracer and interpreter oriented to graph capture for CUDA-graphable training steps. It is a prototype of the design in DESIGN.md. Your code must be CUDA-graphable ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results