未验证 提交 21d4fa02 编写于 作者: Y YuanRisheng 提交者: GitHub

fix onednn bugs, test=document_fix (#48013)

上级 2b81d13c
......@@ -52,7 +52,7 @@ class LayerNormOneDNNHandler
if (scaleshift) {
return this->AcquireMemoryFromPrimitive(
this->fwd_pd_->weights_desc(),
platform::to_void_cast(scaleshift->data<float>()));
phi::funcs::to_void_cast(scaleshift->data<float>()));
} else {
const unsigned int C = phi::vectorize(scale->dims())[0];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册