1 | #pragma once |
---|---|
2 | |
3 | // Test these using #if AT_CUDNN_ENABLED(), not #ifdef, so that it's |
4 | // obvious if you forgot to include Config.h |
5 | // c.f. https://stackoverflow.com/questions/33759787/generating-an-error-if-checked-boolean-macro-is-not-defined |
6 | // |
7 | // NB: This header MUST NOT be included from other headers; it should |
8 | // only be included from C++ files. |
9 | |
10 | #define AT_CUDNN_ENABLED() 1 |
11 | #define AT_ROCM_ENABLED() 0 |
12 | #define AT_MAGMA_ENABLED() 0 |
13 | |
14 | // Needed for hipMAGMA to correctly identify implementation |
15 | #if (AT_ROCM_ENABLED() && AT_MAGMA_ENABLED()) |
16 | #define HAVE_HIP 1 |
17 | #endif |
18 | |
19 | #define NVCC_FLAGS_EXTRA "-gencode;arch=compute_80,code=sm_80" |
20 |