pytorch/torch/csrc/WindowsTorchApiMacro.h
gunandrose4u f07ac6a004 Fix Windows build failure after DDP PR merged (#45335)
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
2020-09-25 12:37:50 -07:00

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