Gao, Xiang 5e97f251a8 Enable TF32 support for cuDNN (#40737) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40737 Reviewed By: mruberry Differential Revision: D22801525 Pulled By: ngimel fbshipit-source-id: ac7f7e728b4b3e01925337e8c9996f26a6433fd2		2020-09-01 15:34:24 -07:00
..
api	Python/C++ API Parity: TransformerDecoder (#42886)	2020-08-28 20:13:53 -07:00
common	Trim libshm deps, move tempfile.h to c10 (#17019)	2019-02-13 19:38:35 -08:00
dist_autograd	Release DistAutogradContainer context for each dist_autograd test case (#38711)	2020-06-18 20:58:55 -07:00
jit	Enable TF32 support for cuDNN (#40737)	2020-09-01 15:34:24 -07:00
rpc	Add set_device_map to TensorPipeOptions to support GPU args (#42637)	2020-08-14 18:46:55 -07:00
tensorexpr	Fix NaN propagation in TE fuser's min/max implementation (#43609)	2020-09-01 02:10:13 -07:00
__init__.py	remediation of S205607	2020-07-17 17:19:47 -07:00