1
2
3
4
5
6
7
8
9
10
11
12
13
| #pragma once
|
| // Test these using #if AT_CUDNN_ENABLED(), not #ifdef, so that it's
| // obvious if you forgot to include Config.h
| // c.f. https://stackoverflow.com/questions/33759787/generating-an-error-if-checked-boolean-macro-is-not-defined
| //
| // NB: This header MUST NOT be included from other headers; it should
| // only be included from C++ files.
|
| #define AT_CUDNN_ENABLED() 1
| #define AT_ROCM_ENABLED() 0
|
| #define NVCC_FLAGS_EXTRA "-gencode;arch=compute_37,code=sm_37;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_61,code=sm_61;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_37,code=compute_37"
|
|
