pytorch/torch/backends
Edward Yang 50df3e5e2e Add ability to query if built with CUDA and MKL-DNN. (#18362)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18362
ghimport-source-id: 374b7ab97e2d6a894368007133201f510539296f

Stack from [ghstack](https://github.com/ezyang/ghstack):
* #18242 Test running a CUDA build on CPU machine.
* **#18362 Add ability to query if built with CUDA and MKL-DNN.**

Fixes #18108.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Differential Revision: D14584430

fbshipit-source-id: 7605a1ac4e8f2a7c70d52e5a43ad7f03f0457473
2019-03-25 10:39:09 -07:00
..
cuda Add ability to query if built with CUDA and MKL-DNN. (#18362) 2019-03-25 10:39:09 -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