p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

clean code 

clean code
This commit is contained in:
jiaorui 2022-09-13 10:33:49 +08:00
parent 9818117c76
commit ab73e9e85f
3 changed files with 53 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
mindspore/ccsrc/plugin/device/ascend/hal/device/kernel_adjust.cc
View File

@ -55,7 +55,7 @@ namespace device {
#ifndef ENABLE_SECURITY
using device::ascend::ProfilingUtils;
#endif
void KernelAdjust::ReorderGetNext(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::ReorderGetNext(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
const std::vector<CNodePtr> &origin_cnode_list = kernel_graph_ptr->execution_order();
std::vector<CNodePtr> getnext_list;
@ -131,7 +131,7 @@ bool KernelAdjust::ExistIndependent(const std::shared_ptr<session::KernelGraph>
void KernelAdjust::InsertIndepentParallel(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
std::vector<CNodePtr> *exec_order) {
std::vector<CNodePtr> *exec_order) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
device::ascend::AscendStreamMng &resource_manager = device::ascend::AscendStreamMng::GetInstance();
@ -149,7 +149,7 @@ void KernelAdjust::InsertIndepentParallel(const std::shared_ptr<session::KernelG
void KernelAdjust::InsertFpBpLoopStreamSwitch(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
std::vector<CNodePtr> *exec_order, uint32_t *fpbp_stream_id,
uint32_t *fpbp_switch_stream_id) {
uint32_t *fpbp_switch_stream_id) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
MS_EXCEPTION_IF_NULL(fpbp_stream_id);
@ -181,7 +181,7 @@ bool KernelAdjust::ExistInitDataSetQueue(const std::shared_ptr<session::KernelGr
});
}
void KernelAdjust::InsertEndGraphTaskSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::InsertEndGraphTaskSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
if (IsTaskSink()) {
auto exec_order = kernel_graph_ptr->execution_order();
@ -197,7 +197,7 @@ void KernelAdjust::InsertEndGraphTaskSink(const std::shared_ptr<session::KernelG
}
void KernelAdjust::InsertEndGraph(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order, uint32_t stream_id) {
std::vector<CNodePtr> *exec_order, uint32_t stream_id) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
CNodePtr end_graph_op = CreateEndGraphOp(kernel_graph_ptr);
@ -209,7 +209,7 @@ void KernelAdjust::InsertEndGraph(const std::shared_ptr<session::KernelGraph> &k
void KernelAdjust::CopyMemcpyList(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::vector<CNodePtr> &orders, size_t order_index,
std::vector<CNodePtr> *memcpy_list, std::vector<CNodePtr> *other_list) {
std::vector<CNodePtr> *memcpy_list, std::vector<CNodePtr> *other_list) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(memcpy_list);
MS_EXCEPTION_IF_NULL(other_list);
@ -243,7 +243,7 @@ void KernelAdjust::InsertEosDoneRecv(const std::shared_ptr<session::KernelGraph>
void KernelAdjust::InsertGetNextLoopStreamActive(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::vector<uint32_t> &getnext_active_streams) {
const std::vector<uint32_t> &getnext_active_streams) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
CNodePtr getnext_active_app = CreateStreamActiveOp(kernel_graph_ptr);
@ -268,7 +268,7 @@ void KernelAdjust::InsertFpBpStartRecv(const std::shared_ptr<session::KernelGrap
void KernelAdjust::InsertNextLoopAssignAdd(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
uint32_t fpbp_stream_id) {
uint32_t fpbp_stream_id) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
CNodePtr assign_add_one = CreateStreamAssignAddnOP(kernel_graph_ptr, switch_loop_input, false);
@ -278,9 +278,9 @@ void KernelAdjust::InsertNextLoopAssignAdd(const std::shared_ptr<session::Kernel
MS_LOG(INFO) << "FpBp loop insert next loop AssignAdd " << assign_add_one->fullname_with_scope();
}
void KernelAdjust::InsertCurrentLoopAssignAdd(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) {
void KernelAdjust::InsertCurrentLoopAssignAdd(
const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, std::vector<CNodePtr> *exec_order,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
CNodePtr cur_assign_add = CreateStreamAssignAddnOP(kernel_graph_ptr, switch_loop_input, true);
@ -292,7 +292,7 @@ void KernelAdjust::InsertCurrentLoopAssignAdd(const std::shared_ptr<session::Ker
void KernelAdjust::InsertFpBpAndEosLoopStreamActive(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::vector<uint32_t> &fpbp_active_streams) {
const std::vector<uint32_t> &fpbp_active_streams) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
CNodePtr fpbp_active_app = CreateStreamActiveOp(kernel_graph_ptr);
@ -306,7 +306,7 @@ void KernelAdjust::InsertFpBpAndEosLoopStreamActive(const std::shared_ptr<sessio
void KernelAdjust::InsertGetNextLoopStreamSwitch(
const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, std::vector<CNodePtr> *exec_order,
uint32_t *getnext_switch_stream_id, uint32_t *getnext_stream_id,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) {
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
MS_EXCEPTION_IF_NULL(getnext_switch_stream_id);
@ -326,7 +326,8 @@ void KernelAdjust::InsertGetNextLoopStreamSwitch(
}
void KernelAdjust::SetBeforeGetNextStreamID(std::vector<CNodePtr> *exec_order, const std::vector<CNodePtr> &orders,
size_t *order_index, CNodePtr getnext_cnode, uint32_t getnext_stream_id) {
size_t *order_index, CNodePtr getnext_cnode,
uint32_t getnext_stream_id) const {
MS_EXCEPTION_IF_NULL(exec_order);
MS_EXCEPTION_IF_NULL(order_index);
for (; *order_index < orders.size(); (*order_index)++) {
@ -371,7 +372,7 @@ void KernelAdjust::InsertGetNextLoopEosStartSend(const std::shared_ptr<session::
void KernelAdjust::InsertEosStreamSwitch(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
std::vector<CNodePtr> *exec_order, uint32_t *eos_switch_stream_id,
uint32_t *eos_stream_id) {
uint32_t *eos_stream_id) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
MS_EXCEPTION_IF_NULL(exec_order);
MS_EXCEPTION_IF_NULL(eos_switch_stream_id);
@ -428,7 +429,7 @@ void KernelAdjust::InsertEosDoneSend(const std::shared_ptr<session::KernelGraph>
MS_LOG(INFO) << "EoS loop insert EoS done Send " << eos_done_send->fullname_with_scope();
}
void KernelAdjust::ProcessLoopSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::ProcessLoopSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
device::ascend::AscendStreamMng &resource_manager = device::ascend::AscendStreamMng::GetInstance();
resource_manager.ResetResource();
@ -539,7 +540,7 @@ kernel::KernelBuildInfo::KernelBuildInfoBuilder KernelAdjust::CreateMngKernelBui
CNodePtr KernelAdjust::CreateStreamSwitchOp(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
StreamSwitchKind kind) {
StreamSwitchKind kind) const {
kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder = CreateMngKernelBuilder(
{kOpFormat_DEFAULT, kOpFormat_DEFAULT}, {TypeId::kNumberTypeInt32, TypeId::kNumberTypeInt32});
auto typeNone_abstract = std::make_shared<abstract::AbstractNone>();
@ -779,7 +780,7 @@ CNodePtr KernelAdjust::CreateNPUClearStatus(const std::shared_ptr<session::Kerne
return npu_clear_cnode;
}
CNodePtr KernelAdjust::CreateNPUAllocStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
CNodePtr KernelAdjust::CreateNPUAllocStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
// create npu_alloc_cnode
auto npu_alloc_primitive = std::make_shared<Primitive>(kNPUAllocFloatStatusOpName);
@ -883,7 +884,7 @@ CNodePtr KernelAdjust::CreateAssign(const std::shared_ptr<session::KernelGraph>
return assign_cnode;
}
void KernelAdjust::InsertOverflowCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::InsertOverflowCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_LOG(INFO) << "Start Insert Overflow Check Operations.";
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
@ -936,7 +937,7 @@ void KernelAdjust::InsertOverflowCheckOperations(const std::shared_ptr<session::
}
void KernelAdjust::InsertGradientOverflowCheckOperations(
const AnfNodePtr &specify_para, const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
const AnfNodePtr &specify_para, const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_LOG(INFO) << "Start Insert Gradient Overflow Check Operations.";
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
@ -994,7 +995,7 @@ void KernelAdjust::InsertGradientOverflowCheckOperations(
}
void KernelAdjust::InsertDynamicLossScaleCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<AnfNodePtr> *dynamic_loss_scale_param_list) {
std::vector<AnfNodePtr> *dynamic_loss_scale_param_list) const {
MS_LOG(INFO) << "Start Insert Dynamic Loss Scale Operations.";
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
auto execution_order = kernel_graph_ptr->execution_order();
@ -1145,7 +1146,7 @@ void KernelAdjust::InsertDeviceLoopCtrl(const std::shared_ptr<session::KernelGra
}
void KernelAdjust::AssignLoopCtrlTensorMem(const session::KernelGraph &kernel_graph, KernelRuntime *runtime_instance,
const string name) {
const string name) const {
MS_EXCEPTION_IF_NULL(runtime_instance);
auto device_loop_control_params = kernel_graph.device_loop_control_params();
if (device_loop_control_params.count(name) == 0) {
@ -1190,7 +1191,7 @@ void KernelAdjust::AssignLoopCtrlTensorMem(const session::KernelGraph &kernel_gr
}
}
void KernelAdjust::AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph_ptr) {
void KernelAdjust::AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph_ptr) const {
auto device_loop_control_tensors = kernel_graph_ptr.device_loop_control_tensors();
if (device_loop_control_tensors.empty()) {
return;
@ -1209,7 +1210,7 @@ void KernelAdjust::AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph
}
void KernelAdjust::SetDeviceLoopCtrlTensor(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::string name, int32_t value) {
const std::string name, int32_t value) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
auto device_loop_control_tensors = kernel_graph_ptr->device_loop_control_tensors();
if (device_loop_control_tensors.count(name) == 0) {
@ -1230,7 +1231,7 @@ void KernelAdjust::SetDeviceLoopCtrlTensor(const std::shared_ptr<session::Kernel
}
}
void KernelAdjust::LoadDeviceLoopCtrlParameters(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::LoadDeviceLoopCtrlParameters(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
MS_EXCEPTION_IF_NULL(kernel_graph_ptr);
auto device_loop_control_tensors = kernel_graph_ptr->device_loop_control_tensors();
if (device_loop_control_tensors.empty()) {

46
mindspore/ccsrc/plugin/device/ascend/hal/device/kernel_adjust.h
View File

@ -60,11 +60,11 @@ class KernelAdjust {
}
// device loop control
void InsertDeviceLoopCtrl(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
void AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph_ptr);
void LoadDeviceLoopCtrlParameters(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
void AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph_ptr) const;
void LoadDeviceLoopCtrlParameters(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
void InsertOverflowCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
void ProcessLoopSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
void InsertOverflowCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
void ProcessLoopSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
#ifndef ENABLE_SECURITY
void Profiling(NotNull<session::KernelGraph *> kernel_graph_ptr);
#endif
@ -82,15 +82,15 @@ class KernelAdjust {
const CNodePtr &npu_alloc_cnode) const;
CNodePtr CreateNPUClearStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const CNodePtr &npu_alloc_cnode) const;
CNodePtr CreateNPUAllocStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
CNodePtr CreateNPUAllocStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
CNodePtr CreateAssignAdd(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const CNodePtr &npu_alloc_cnode, const AnfNodePtr &specify_para) const;
CNodePtr CreateAssign(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const AnfNodePtr &specify_para) const;
void ReorderGetNext(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
void ReorderGetNext(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
CNodePtr CreateStreamSwitchOp(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
StreamSwitchKind kind);
StreamSwitchKind kind) const;
CNodePtr CreateEndGraphOp(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
CNodePtr CreatTupleGetItemNode(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, const CNodePtr &node,
size_t output_idx) const;
@ -111,9 +111,9 @@ class KernelAdjust {
void InsertGetNextLoopStreamSwitch(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order, uint32_t *getnext_switch_stream_id,
uint32_t *getnext_stream_id,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input);
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) const;
void SetBeforeGetNextStreamID(std::vector<CNodePtr> *exec_order, const std::vector<CNodePtr> &orders,
size_t *order_index, CNodePtr getnext_cnode, uint32_t getnext_stream_id);
size_t *order_index, CNodePtr getnext_cnode, uint32_t getnext_stream_id) const;
void InsertGetNextLoopFpBpStartSend(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order, uint32_t *fpbp_start_event_id,
uint32_t getnext_stream_id) const;
@ -123,7 +123,7 @@ class KernelAdjust {
void InsertEosStreamSwitch(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
std::vector<CNodePtr> *exec_order, uint32_t *eos_switch_stream_id,
uint32_t *eos_stream_id);
uint32_t *eos_stream_id) const;
void InsertGetNextLoopEosStartRecv(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order, uint32_t eos_start_event_id,
uint32_t eos_stream_id) const;
@ -133,43 +133,43 @@ class KernelAdjust {
std::vector<CNodePtr> *exec_order, uint32_t *eos_done_event_id, uint32_t eos_stream_id) const;
void InsertIndepentParallel(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
std::vector<CNodePtr> *exec_order);
std::vector<CNodePtr> *exec_order) const;
void InsertFpBpLoopStreamSwitch(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
std::vector<CNodePtr> *exec_order, uint32_t *fpbp_stream_id,
uint32_t *fpbp_switch_stream_id);
void InsertEndGraphTaskSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
uint32_t *fpbp_switch_stream_id) const;
void InsertEndGraphTaskSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
void InsertEndGraph(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, std::vector<CNodePtr> *exec_order,
uint32_t stream_id);
uint32_t stream_id) const;
void InsertFpBpStartRecv(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order, uint32_t fpbp_start_event_id,
uint32_t fpbp_stream_id) const;
void InsertNextLoopAssignAdd(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input,
uint32_t fpbp_stream_id);
uint32_t fpbp_stream_id) const;
void CopyMemcpyList(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const std::vector<CNodePtr> &orders, size_t order_index, std::vector<CNodePtr> *memcpy_list,
std::vector<CNodePtr> *other_list);
std::vector<CNodePtr> *other_list) const;
void InsertEosDoneRecv(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order, uint32_t eos_done_event_id, uint32_t fpbp_stream_id) const;
void InsertGetNextLoopStreamActive(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::vector<uint32_t> &getnext_active_streams);
const std::vector<uint32_t> &getnext_active_streams) const;
void InsertCurrentLoopAssignAdd(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input);
const std::map<std::string, mindspore::ParameterPtr> &switch_loop_input) const;
void InsertFpBpAndEosLoopStreamActive(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<CNodePtr> *exec_order,
const std::vector<uint32_t> &fpbp_active_streams);
const std::vector<uint32_t> &fpbp_active_streams) const;
void SetDeviceLoopCtrlTensor(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, const string name,
int32_t value);
int32_t value) const;
void AssignLoopCtrlTensorMem(const session::KernelGraph &kernel_graph, KernelRuntime *runtime_instance,
const string name);
const string name) const;
void InsertGradientOverflowCheckOperations(const AnfNodePtr &specify_para,
const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const;
void InsertDynamicLossScaleCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
std::vector<AnfNodePtr> *dynamic_loss_scale_param_list);
std::vector<AnfNodePtr> *dynamic_loss_scale_param_list) const;
std::shared_ptr<Tensor> CreateTensor(int32_t initial_value) const;
std::shared_ptr<Parameter> CreateParameter(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
const string parameter_name) const;

8
tests/ut/cpp/stub/tasksink/ascend_stream_assign_stub.cc
View File

@ -36,19 +36,19 @@ uint32_t AscendStreamAssign::GetHcomTaskNum(const CNodePtr &) { return 200; }
} // namespace ascend
void KernelAdjust::InsertDeviceLoopCtrl(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const { return; }
void KernelAdjust::AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph_ptr) { return; }
void KernelAdjust::LoadDeviceLoopCtrlParameters(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::AssignLoopCtrlMemory(const session::KernelGraph &kernel_graph_ptr) const { return; }
void KernelAdjust::LoadDeviceLoopCtrlParameters(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
return;
}
bool KernelAdjust::NeedLoopSink() { return true; }
void KernelAdjust::ProcessLoopSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) { return; }
void KernelAdjust::ProcessLoopSink(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const { return; }
#ifndef ENABLE_SECURITY
void KernelAdjust::Profiling(NotNull<session::KernelGraph *> kernel_graph_ptr) { return; }
#endif
void KernelAdjust::InsertOverflowCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) {
void KernelAdjust::InsertOverflowCheckOperations(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr) const {
return;
}
} // namespace device