提交 bebd5c8c 编写于 作者: S superjomn

update

上级 d911c343
...@@ -31,7 +31,7 @@ class SGDCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> { ...@@ -31,7 +31,7 @@ class SGDCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> {
void Run() override { void Run() override {
auto &context = ctx_->As<X86Context>(); auto &context = ctx_->As<X86Context>();
auto &sgd_param = *param_.get_mutable<operators::SGDParam>(); auto &sgd_param = *param_.get_mutable<operators::SGDParam>();
CHECK(context.x86_device_context); CHECK(context.x86_device_context());
// param.Out->template mutable_data<T>(); // param.Out->template mutable_data<T>();
...@@ -50,7 +50,7 @@ class SGDCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> { ...@@ -50,7 +50,7 @@ class SGDCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> {
const T *grad_data = grad->data<T>(); const T *grad_data = grad->data<T>();
int64_t rows_idx = 0; int64_t rows_idx = 0;
T *out_data = T *out_data =
param_out->mutable_data<T>(context.x86_device_context->GetPlace()); param_out->mutable_data<T>(context.x86_device_context()->GetPlace());
auto sgd = auto sgd =
paddle::operators::jit::KernelFuncs<paddle::operators::jit::SgdTuple<T>, paddle::operators::jit::KernelFuncs<paddle::operators::jit::SgdTuple<T>,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册