Browse Source

fix

pull/2052/head
guopeian 3 years ago
parent
commit
0f2fe9b165
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/single_op/task/op_task.h

+ 1
- 0
ge/single_op/task/op_task.h View File

@@ -172,6 +172,7 @@ class AiCpuBaseTask : public OpTask {
std::vector<GeTensorDesc> &output_desc,
std::vector<DataBuffer> &output_buffers,
rtStream_t stream) override;
virtual Status LaunchKernel(rtStream_t stream) = 0;
protected:
Status UpdateIoAddr(const std::vector<DataBuffer> &inputs, const std::vector<DataBuffer> &outputs);
Status SetInputConst();


Loading…
Cancel
Save