From 3773260edde142a194a590b79f694a446f1b8cd9 Mon Sep 17 00:00:00 2001
From: lujun <lujun315023@126.com>
Date: Tue, 2 Apr 2019 16:18:41 +0800
Subject: [PATCH] fix load type, test=develop

---
 paddle/fluid/operators/load_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/load_op.cc b/paddle/fluid/operators/load_op.cc
index 656728c609..435c755df3 100644
--- a/paddle/fluid/operators/load_op.cc
+++ b/paddle/fluid/operators/load_op.cc
@@ -29,7 +29,7 @@ class LoadOp : public framework::OperatorWithKernel {
   framework::OpKernelType GetExpectedKernelType(
       const framework::ExecutionContext &ctx) const override {
     framework::OpKernelType kt = framework::OpKernelType(
-        framework::proto::VarType::FP32, platform::CPUPlace());
+        framework::proto::VarType::FP32, ctx.GetPlace());
     return kt;
   }
 };
-- 
GitLab