提交 3c84ebec 编写于 作者: Y yangyaming

IndicateDataType --> GetKernelType

上级 40367d18
...@@ -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.
先完成此消息的编辑!
想要评论请 注册