pytorch
4372dbb8 - use pytree to allow any input format for cuda graph (#90941)

Commit
2 years ago
use pytree to allow any input format for cuda graph (#90941) Summary: 1. use pytree to allow any input format for make_graphed_callables 2. add allow_unused_input argument for make_graphed_callables Test Plan: buck2 test mode/dev-nosan //caffe2/test:cuda -- --print-passing-details Differential Revision: D42077976 Pull Request resolved: https://github.com/pytorch/pytorch/pull/90941 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading