commit bcba5a50d1b3444bf882f8a861b667c3328aed44
author Jerry Zhang <jerryzh@fb.com> Fri Jul 27 21:24:48 2018 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Fri Jul 27 22:01:23 2018 -0700
tree f9e20767df3ba5e9ec889270e24160417d959e06
parent ab4e2090079c85f48019fa2f21027eb6571fb30f

Fix EnforceFiniteOp

Summary: att

Reviewed By: kennyhorror

Differential Revision: D9040248

fbshipit-source-id: 0da0f3b1ce51375731098cc86c92f35953be0861

caffe2/operators/enforce_finite_op.h

diff --git a/caffe2/operators/enforce_finite_op.h b/caffe2/operators/enforce_finite_op.h
index a1f63ec..19fe96b 100644
--- a/caffe2/operators/enforce_finite_op.h
+++ b/caffe2/operators/enforce_finite_op.h
@@ -23,7 +23,7 @@
   bool DoRunWithType();
 
  private:
-  Tensor buffer_{Context::GetDeviceType()};
+  Tensor buffer_{CPU};
 
   template <typename T>
   void EnforceOnCPU(const Tensor& input) {