mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
dc186cc9fe
* remove legacy options from CMakeLists * codemod WITH_ to USE_ for WITH_CUDA, WITH_CUDNN, WITH_DISTRIBUTED, WITH_DISTRIBUTED_MW, WITH_GLOO_IBVERBS, WITH_NCCL, WITH_ROCM, WITH_NUMPY * cover SYSTEM_NCCL, MKLDNN, NNPACK, C10D, NINJA * removed NO_* variables and hotpatch them only in setup.py * fix lint
17 lines
303 B
C
17 lines
303 B
C
#ifndef THP_API_H
|
|
#define THP_API_H
|
|
|
|
#ifdef _THP_CORE
|
|
#error Using the THP API header, but _THP_CORE is defined! This macro should \
|
|
be defined only when compiling the core torch package.
|
|
#endif
|
|
|
|
#ifdef USE_CUDA
|
|
#include "cuda/THCP.h"
|
|
#include "cuda/undef_macros.h"
|
|
#endif
|
|
|
|
#include "THP.h"
|
|
|
|
#endif
|