Created by: Shixiaowei02
1、对齐 Lite 与 Fluid 的 framework.proto; 2、添加设定算子版本的接口,算子版本号与 Fluid 主框架版本号格式一致(三位小数组成的字符串,如 "1.5.0"),在注册时进行设定。示例:
REGISTER_LITE_KERNEL(leaky_relu,
kCUDA,
kFloat,
kNCHW,
paddle::lite::kernels::cuda::LeakyReluCompute,
def)
.BindInput("X", {LiteType::GetTensorTy(TARGET(kCUDA))})
.BindOutput("Out", {LiteType::GetTensorTy(TARGET(kCUDA))})
.SetVersion("1.5.0")
.Finalize();
注意:上述接口注册的算子版本是 Kernel 级别的,也就是说,同一名称的 Operator(如 leaky_relu)在不同的 Place(如 kARM 和 kCUDA)下,版本号是分开注册、可以不同的。