Add output defs for sgd kernel #51332
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
|
Could you give some comments? @From00 |
| @@ -187,7 +187,9 @@ PD_REGISTER_KERNEL(sgd, | |||
| phi::SGDDenseKernel, | |||
| phi::dtype::float16, | |||
| float, | |||
| double) {} | |||
| double) { | |||
| kernel->OutputAt(1).SetDataType(phi::DataType::UNDEFINED); | |||
There was a problem hiding this comment.
master_param_out注册UNDEFINED后，你需要在SgdInferMeta中对其数据类型进行推导。
There was a problem hiding this comment.
根据
using MPDType = typename phi::dtype::MPTypeTrait<T>::Type;你需要在InferMeta中实现和phi::dtype::MPTypeTrait<T>::Type等价的类型推导逻辑。请查看phi::dtype::MPTypeTrait<T>::Type，根据模板类型推断实现InferMeta中的等价判断逻辑。
There was a problem hiding this comment.
根据
using MPDType = typename phi::dtype::MPTypeTrait<T>::Type;你需要在InferMeta中实现和
phi::dtype::MPTypeTrait<T>::Type等价的类型推导逻辑。请查看phi::dtype::MPTypeTrait<T>::Type，根据模板类型推断实现InferMeta中的等价判断逻辑。
谢谢回复，这里有点不明白，是可以抄这个模板推断还是用一串if else给出不同的类型时的结果。
There was a problem hiding this comment.
用if-else实现，InferMeta中无法获取模板类型进行推断，你需要根据输入的dtype，判断出不同的输出结果。
PR types
Others
PR changes
Others
Describe
Task 59 in #51292