Tongzhou Wang 973d51079b Add device-specific cuFFT plan caches (#19300) ... Summary: Fixes https://github.com/pytorch/pytorch/issues/19224 Pull Request resolved: https://github.com/pytorch/pytorch/pull/19300 Differential Revision: D14986967 Pulled By: soumith fbshipit-source-id: 8c31237db50d6924bba1472434c10326610d9255		2019-04-18 06:39:35 -07:00
..
cuda	Add device-specific cuFFT plan caches (#19300)	2019-04-18 06:39:35 -07:00
cudnn	Fix the flake8 linter	2019-01-30 09:36:00 -08:00
mkl	[fft][1 of 3] build system and helpers to support cuFFT and MKL (#5855)	2018-03-19 15:43:14 -04:00
mkldnn	Add ability to query if built with CUDA and MKL-DNN. (#18362)	2019-03-25 10:39:09 -07:00
openmp	Add torch.backends.openmp.is_available(); fix some cmake messages (#16425)	2019-01-31 16:15:46 -08:00
__init__.py	Add cuDNN support for convolutions (#36)	2016-09-27 17:55:04 -04:00