Xiang Gao 23174ca71b [reland] Enable TF32 support for cuBLAS (#41498) ... Summary: fix rocm Pull Request resolved: https://github.com/pytorch/pytorch/pull/41498 Reviewed By: mruberry Differential Revision: D22560572 Pulled By: ngimel fbshipit-source-id: 5ee79e96cb29e70d9180830d058efb53d1c6c041		2020-07-15 21:00:55 -07:00
..
cuda	[reland] Enable TF32 support for cuBLAS (#41498)	2020-07-15 21:00:55 -07:00
cudnn	Fix torch.backends.cudnn mypy error (#38947)	2020-06-03 10:55:43 -07:00
mkl
mkldnn	Add torch.backends.mkldnn.enabled flag (#25459)	2019-09-11 12:09:40 -07:00
openmp
quantized	Remove fbgemm_is_cpu_supported in favor of torch.backends.quantized.supported_qengines (#26840)	2019-09-27 13:45:15 -07:00
xnnpack	Integrate XNNPACK with custom class for packing weights. (#34047)	2020-03-14 12:51:56 -07:00
__init__.py	Add torch.backends.mkldnn.enabled flag (#25459)	2019-09-11 12:09:40 -07:00