mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
Summary:
Fixes #{issue number}
This is resubmit for PR https://github.com/pytorch/pytorch/issues/42897 . Together with fix for Windows build issue introduced by PR https://github.com/pytorch/pytorch/issues/44344 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45335
Reviewed By: zou3519
Differential Revision: D23931471
Pulled By: mrshenli
fbshipit-source-id: f49b5a114944c1450b32934b3292170be064f494
14 lines
279 B
C
14 lines
279 B
C
#pragma once
|
|
|
|
#include <c10/macros/Export.h>
|
|
|
|
// There's no difference between aten, torch and caffe2 libs any more
|
|
// TODO: clean up the naming for consistency
|
|
#define TORCH_API CAFFE2_API
|
|
|
|
#ifdef _WIN32
|
|
#define TORCH_PYTHON_API
|
|
#else
|
|
#define TORCH_PYTHON_API CAFFE2_API
|
|
#endif
|