diff --git a/ge/single_op/task/op_task.h b/ge/single_op/task/op_task.h index a0d5b574..ac7e489c 100644 --- a/ge/single_op/task/op_task.h +++ b/ge/single_op/task/op_task.h @@ -172,6 +172,7 @@ class AiCpuBaseTask : public OpTask { std::vector &output_desc, std::vector &output_buffers, rtStream_t stream) override; + virtual Status LaunchKernel(rtStream_t stream) = 0; protected: Status UpdateIoAddr(const std::vector &inputs, const std::vector &outputs); Status SetInputConst();