Fix linking errors when building dataloader test binaries on Windows (#17494)
authorpeter <peterghost86@gmail.com>
Wed, 27 Feb 2019 04:33:59 +0000 (20:33 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 27 Feb 2019 04:36:45 +0000 (20:36 -0800)
Summary:
Fixes #17489.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17494

Differential Revision: D14226525

Pulled By: ezyang

fbshipit-source-id: 3dfef9bc6f443d647e9f05a54bc17c5717033723

torch/csrc/api/include/torch/data/samplers/distributed.h

index ff2ade0..2289915 100644 (file)
@@ -38,11 +38,11 @@ class DistributedSampler : public Sampler<BatchRequest> {
 
   /// Set the epoch for the current enumeration. This can be used to alter the
   /// sample selection and shuffling behavior.
-  TORCH_API void set_epoch(size_t epoch) {
+  void set_epoch(size_t epoch) {
     epoch_ = epoch;
   }
 
-  TORCH_API size_t epoch() const {
+  size_t epoch() const {
     return epoch_;
   }