51 lines
1.5 KiB
C
51 lines
1.5 KiB
C
|
/* Copyright 2022 The ml_dtypes Authors.
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License.
|
||
|
|
==============================================================================*/
|
||
|
|
|
||
|
|
#ifndef ML_DTYPES__NUMPY_H_
|
||
|
|
#define ML_DTYPES__NUMPY_H_
|
||
|
|
|
||
|
|
#ifdef PyArray_Type
|
||
|
|
#error "Numpy cannot be included before numpy.h."
|
||
|
|
#endif
|
||
|
|
|
||
|
|
// Disallow Numpy 1.7 deprecated symbols.
|
||
|
|
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
|
||
|
|
|
||
|
|
// We import_array in the ml_dtypes init function only.
|
||
|
|
#define PY_ARRAY_UNIQUE_SYMBOL _ml_dtypes_numpy_api
|
||
|
|
#ifndef ML_DTYPES_IMPORT_NUMPY
|
||
|
|
#define NO_IMPORT_ARRAY
|
||
|
|
#endif
|
||
|
|
|
||
|
|
// Place `<locale>` before <Python.h> to avoid build failure in macOS.
|
||
|
|
#include <locale>
|
||
|
|
|
||
|
|
#include <Python.h>
|
||
|
|
|
||
|
|
#include "numpy/arrayobject.h"
|
||
|
|
#include "numpy/ufuncobject.h"
|
||
|
|
|
||
|
|
namespace ml_dtypes {
|
||
|
|
|
||
|
|
// Import numpy. This wrapper function exists so that the
|
||
|
|
// PY_ARRAY_UNIQUE_SYMBOL can be safely defined in a .cc file to
|
||
|
|
// avoid weird linking issues. Should be called only from our
|
||
|
|
// module initialization function.
|
||
|
|
void ImportNumpy();
|
||
|
|
|
||
|
|
} // namespace ml_dtypes
|
||
|
|
|
||
|
|
#endif // ML_DTYPES__NUMPY_H_
|