pytorch/torch/backends
Jithun Nair 718c538ff9 Add ability to enable/disable MIOpen at runtime (#33118)
Summary:
1. Set `torch._C.has_cudnn` to `True` for ROCm
2. Make MIOpen invocations respect value of `cudnn_enabled` or `at::globalContext().userEnabledCuDNN()`
3. `torch/backends/cudnn/__init__.py`: Add hip-specific changes (use "hide whitespace changes" option to view simpler diff)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33118

Differential Revision: D19977719

Pulled By: bddppq

fbshipit-source-id: 64d4dd1d78afcf96201360d85b8be5950f96dfad
2020-02-20 10:47:57 -08:00
..
cuda Add device-specific cuFFT plan caches (#19300) 2019-04-18 06:39:35 -07:00
cudnn Add ability to enable/disable MIOpen at runtime (#33118) 2020-02-20 10:47:57 -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 torch.backends.mkldnn.enabled flag (#25459) 2019-09-11 12:09:40 -07:00
openmp Add torch.backends.openmp.is_available(); fix some cmake messages (#16425) 2019-01-31 16:15:46 -08:00
quantized Remove fbgemm_is_cpu_supported in favor of torch.backends.quantized.supported_qengines (#26840) 2019-09-27 13:45:15 -07:00
__init__.py Add torch.backends.mkldnn.enabled flag (#25459) 2019-09-11 12:09:40 -07:00