Add ability to query if built with CUDA and MKL-DNN. (#18362)
authorEdward Yang <ezyang@fb.com>
Mon, 25 Mar 2019 17:22:54 +0000 (10:22 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 25 Mar 2019 17:39:09 +0000 (10:39 -0700)
commit50df3e5e2e2c51db6389b5c47c048f9db112c5c3
tree00db997e212da7921ad6b50e6eb3c65a19c57e0f
parent17fcdfb9257c1fd2e594c2f30abd8632b3e6b7b2
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
aten/src/ATen/Context.cpp
aten/src/ATen/Context.h
test/test_torch.py
torch/backends/cuda/__init__.py
torch/backends/mkldnn/__init__.py [new file with mode: 0644]
torch/csrc/Module.cpp