pytorch
66b20bb7 - [CUDA graphs] [JIT] improves readability and nvfuser convenience for graph-safe cuda RNG (#51580)

Commit
3 years ago
[CUDA graphs] [JIT] improves readability and nvfuser convenience for graph-safe cuda RNG (#51580) Summary: I'm trying to make jitted RNG graph-safe in csarofeen 's nvfuser branch. Doing so requires diffs in files outside torch/csrc/jit, and we'd like these to go upstream through the present simple separate PR (instead of needing to be reviewed as part of Christian's branch's eventual merge, which will be massive). From the perspective of eager mode consumers, diffs here are purely cosmetic. I moved raw definitions of `PhiloxCudaState` and `at::cuda::philox::unpack` to standalone headers the codegen can easily copy from. Pull Request resolved: https://github.com/pytorch/pytorch/pull/51580 Reviewed By: malfet Differential Revision: D26626972 Pulled By: ngimel fbshipit-source-id: 7f04d6c5ffe0af7a8a66d3ae6ed36191d12f7d67
Author
Parents
Loading