pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00

History

Michael Carilli 929f1d5317 [RELAND] Adds torch.cuda.is_current_stream_capturing (#77789 ) Resubmit of https://github.com/pytorch/pytorch/pull/77673, which was reverted due to Windows test failures: https://github.com/pytorch/pytorch/pull/77673#issuecomment-1130425845. I suspect these failures happened because I don't explicitly set a side stream for graph capture in the new test. Not setting a side stream explicitly is alright on Linux because cuda tests implicitly use a side stream. I think Windows cuda tests implicitly use the default stream, breaking capture and leaving the backend in a bad state. Other graphs tests explicitly set side streams and don't error in Windows builds, so i'm 95% sure doing the same for the new test will work. Pull Request resolved: https://github.com/pytorch/pytorch/pull/77789 Approved by: https://github.com/ezyang		2022-05-18 23:18:53 +00:00
..
caffe2
cpp	Revert "Revert "Allow specifying tags for aten operators in native_functions.yaml""	2022-04-28 02:04:57 +00:00
source	[RELAND] Adds torch.cuda.is_current_stream_capturing (#77789 )	2022-05-18 23:18:53 +00:00
.gitignore
libtorch.rst
make.bat
Makefile	Add list of supported ATen ops by ONNX converter into torch.onnx page	2022-04-07 00:05:44 +00:00
README.md
requirements.txt	Jinja2 version pinned to 3.0.* (#74690 )	2022-03-24 21:58:28 +00:00

Please see the Writing documentation section of CONTRIBUTING.md for details on both writing and building the docs.