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

IndicateDataType --> GetKernelType

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