Browse Source

fix sc +

tags/v1.5.1
wangxiaotian22 3 years ago
parent
commit
e7296ed73c
2 changed files with 6 additions and 2 deletions
  1. +1
    -1
      ge/ge_runtime/task/label_goto_task.cc
  2. +5
    -1
      ge/single_op/task/op_task.h

+ 1
- 1
ge/ge_runtime/task/label_goto_task.cc View File

@@ -72,7 +72,7 @@ bool LabelGotoTask::Distribute() {
return false; return false;
} }


rt_ret = rtLabelListCpy((void**)label_list.data(), label_list.size(), label_info_, label_info_size);
rt_ret = rtLabelListCpy(reinterpret_cast<void**>(label_list.data()), label_list.size(), label_info_, label_info_size);
if (rt_ret != RT_ERROR_NONE) { if (rt_ret != RT_ERROR_NONE) {
GELOGE(RT_FAILED, "Call rt api failed, ret: %#x", rt_ret); GELOGE(RT_FAILED, "Call rt api failed, ret: %#x", rt_ret);
return false; return false;


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

@@ -33,6 +33,10 @@
#include "register/op_tiling.h" #include "register/op_tiling.h"


namespace ge { namespace ge {
namespace {
const int kAddressNum = 2;
} // namespace

class StreamResource; class StreamResource;
struct SingleOpModelParam; struct SingleOpModelParam;
class OpTask { class OpTask {
@@ -256,7 +260,7 @@ class MemcpyAsyncTask : public OpTask {
friend class SingleOpModel; friend class SingleOpModel;
friend class RtsKernelTaskBuilder; friend class RtsKernelTaskBuilder;


uintptr_t addresses_[2];
uintptr_t addresses_[kAddressNum];
size_t dst_max_; size_t dst_max_;
size_t count_; size_t count_;
rtMemcpyKind_t kind_; rtMemcpyKind_t kind_;


Loading…
Cancel
Save