Merge pull request #16634 from junjun315/my-cool-stuff

fix load bug about context.place
revert-16807-engine2-interface
lujun 6 years ago committed by GitHub
commit 6173585051
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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;
}
};

Loading…
Cancel
Save