Export generator constructor (#14041)
authorZeming Lin <ebetica0@gmail.com>
Tue, 4 Dec 2018 21:43:28 +0000 (13:43 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 4 Dec 2018 21:50:06 +0000 (13:50 -0800)
Summary:
Missed a spot :)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14041

Reviewed By: ezyang

Differential Revision: D13283803

Pulled By: ebetica

fbshipit-source-id: 482e245f57b0cea6ca3886355ea3ae487d024d4b

aten/src/ATen/templates/GeneratorDerived.h

index 50d7276..d3288f3 100644 (file)
@@ -10,7 +10,7 @@ namespace at {
 
 class Context;
 struct ${name}Generator : public Generator {
-  ${name}Generator(Context * context);
+  CAFFE2_API ${name}Generator(Context * context);
   CAFFE2_API virtual ~${name}Generator();
 
   CAFFE2_API virtual ${name}Generator& copy(const Generator& from) override;