mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
24314e9ceb
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15769 Reviewed By: yf225, pjh5 Differential Revision: D13597845 Pulled By: orionr fbshipit-source-id: 99640e22974990ae570a4795ce07274c4447cb01
22 lines
391 B
C
22 lines
391 B
C
#pragma once
|
|
|
|
#ifdef _WIN32
|
|
#if !defined(TORCH_BUILD_STATIC_LIBS)
|
|
#if defined(torch_EXPORTS)
|
|
#define TORCH_API __declspec(dllexport)
|
|
#else
|
|
#define TORCH_API __declspec(dllimport)
|
|
#endif
|
|
#else
|
|
#define TORCH_API
|
|
#endif
|
|
#elif defined(__GNUC__)
|
|
#if defined(torch_EXPORTS)
|
|
#define TORCH_API __attribute__((__visibility__("default")))
|
|
#else
|
|
#define TORCH_API
|
|
#endif
|
|
#else
|
|
#define TORCH_API
|
|
#endif
|