mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a5d356cb39
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29143 THP_CORE macro is a very old macro that appeared to have served two purposes: 1. The torch-python equivalent of CAFFE2_BUILD_MAIN_LIB, to toggle symbol visibility headers 2. Some sort of ad hoc way of hiding certain definitions from headers so external clients can't get at them. It did (2) in a very confusing manner, because we set THP_CORE in both torch and torch-python (it shouldn't do anything in torch). In this PR I just get rid of use case (2) entirely (so everything shows up in headers all the time), and then redo (1) using a new THP_BUILD_MAIN_LIB macro. This cleans up some of the macro definitions and makes my life easier for working on #27215. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Test Plan: Imported from OSS Differential Revision: D18309594 Pulled By: ezyang fbshipit-source-id: adcb6d7cb387cd818480137e2b94e5e761dbfefc
15 lines
370 B
C++
15 lines
370 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/python_headers.h>
|
|
#include <torch/csrc/autograd/variable.h>
|
|
#include <cstdint>
|
|
|
|
extern PyTypeObject THPSizeType;
|
|
|
|
#define THPSize_Check(obj) (Py_TYPE(obj) == &THPSizeType)
|
|
|
|
PyObject * THPSize_New(const torch::autograd::Variable& t);
|
|
PyObject * THPSize_NewFromSizes(int dim, const int64_t *sizes);
|
|
|
|
void THPSize_init(PyObject *module);
|