1#pragma once
2
3#include <c10/macros/Export.h>
4
5#ifdef THP_BUILD_MAIN_LIB
6#define TORCH_PYTHON_API C10_EXPORT
7#else
8#define TORCH_PYTHON_API C10_IMPORT
9#endif
10