mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +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
22 lines
399 B
C
22 lines
399 B
C
#pragma once
|
|
|
|
#include "torch/csrc/python_headers.h"
|
|
|
|
#ifdef USE_NUMPY
|
|
|
|
#if !defined(NO_IMPORT_ARRAY) && !defined(WITH_NUMPY_IMPORT_ARRAY)
|
|
#define NO_IMPORT_ARRAY
|
|
#endif
|
|
|
|
#ifndef PY_ARRAY_UNIQUE_SYMBOL
|
|
#define PY_ARRAY_UNIQUE_SYMBOL __numpy_array_api
|
|
#endif
|
|
|
|
#ifndef NPY_NO_DEPRECATED_API
|
|
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
|
|
#endif
|
|
|
|
#include <numpy/arrayobject.h>
|
|
|
|
#endif // USE_NUMPY
|