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

Merge pull request #5542 from pkuyym/fix-5535

IndicateDataType --> GetKernelType
...@@ -45,9 +45,10 @@ class ChunkEvalOp : public framework::OperatorWithKernel { ...@@ -45,9 +45,10 @@ class ChunkEvalOp : public framework::OperatorWithKernel {
} }
protected: protected:
framework::DataType IndicateDataType( framework::OpKernelType GetKernelType(
const framework::ExecutionContext &ctx) const override { 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.
先完成此消息的编辑!
想要评论请 注册