未验证 提交 bb4b9af7 编写于 作者: Q Qiao Longfei 提交者: GitHub

Merge pull request #10056 from typhoonzero/fix_splitbyref_macbuild

fix split byref op mac build
...@@ -33,7 +33,7 @@ class SplitByrefOpKernel : public framework::OpKernel<T> { ...@@ -33,7 +33,7 @@ class SplitByrefOpKernel : public framework::OpKernel<T> {
// NOTE: no need to call mutable_data here to allocate memory. // NOTE: no need to call mutable_data here to allocate memory.
auto* out = outs[i]; auto* out = outs[i];
VLOG(3) << "spliting by ref: " << row_offset << " " << out->dims()[0]; VLOG(3) << "spliting by ref: " << row_offset << " " << out->dims()[0];
*out = std::move(in->Slice(row_offset, row_offset + out->dims()[0])); *out = in->Slice(row_offset, row_offset + out->dims()[0]);
row_offset += out->dims()[0]; row_offset += out->dims()[0];
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册