• C
    improve CinnLaunchOpKernel implement (#36936) · 0590277a
    CtfGo 提交于
    1. 功能不变,简化CinnLaunchOpKernel实现:将原先直接从Scope获取变量信息的方式改为借助参数ExecutionContext标准接口获取,简化了实现逻辑,相应地也简化了辅助函数的实现,原先cinn_launch_op_helper较为冗余,删除不必要的接口并迁移至cinn_launch_op.cc中定义。
    2. 修复CinnLaunchOp InferShape判断是否有指定输出:HasOutput->HasOutputs
    3. 添加详细的注释和debug信息,方便问题排查和代码维护
    0590277a
cinn_launch_op_test.cc 11.5 KB