未验证 提交 2b201889 编写于 作者: Y Yang yaming 提交者: GitHub

Merge pull request #5542 from pkuyym/fix-5535

IndicateDataType --> GetKernelType
......@@ -45,9 +45,10 @@ class ChunkEvalOp : public framework::OperatorWithKernel {
}
protected:
framework::DataType IndicateDataType(
framework::OpKernelType GetKernelType(
const framework::ExecutionContext &ctx) const override {
return framework::DataType::FP32;
return framework::OpKernelType(framework::DataType::FP32,
ctx.device_context());
}
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册