| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 114 struct KernelBuildTaskInfo task_info; in SavePreBuildTaskInfo() local
115 task_info.node = anf_node; in SavePreBuildTaskInfo()
116 task_info.json_name = json_name; in SavePreBuildTaskInfo()
118 task_info.processor = tbe::kProcessorAiCore; in SavePreBuildTaskInfo()
120 task_info.processor = tbe::GetProcessor(anf_node); in SavePreBuildTaskInfo()
122 task_info.scope_id = 0; in SavePreBuildTaskInfo()
123 pre_build_task_map_[task_id] = task_info; in SavePreBuildTaskInfo()
130 struct KernelBuildTaskInfo task_info; in SaveTaskInfo() local
131 task_info.node = anf_node; in SaveTaskInfo()
132 task_info.json_name = json_name; in SaveTaskInfo()
[all …]
|
| D | ascend_kernel_compile.cc | 318 …dKernelCompileManager::QueryResultProcess(const nlohmann::json &json, TargetJobStatus *task_info) { in QueryResultProcess() argument
331 task_info->target_job_id = target_job_id; in QueryResultProcess()
332 task_info->json_name = json_name; in QueryResultProcess()
333 task_info->except_msg = message; in QueryResultProcess()
335 task_info->job_status = kSuccess; in QueryResultProcess()
337 task_info->job_status = kFailed; in QueryResultProcess()
364 struct TargetJobStatus task_info; in ParseTargetJobStatus() local
365 QueryResultProcess(json_obj, &task_info); in ParseTargetJobStatus()
366 auto target_node = job_id_to_node_[task_info.target_job_id]; in ParseTargetJobStatus()
367 if (task_info.job_status == kSuccess) { in ParseTargetJobStatus()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | stream_switch_task.cc | 24 const std::shared_ptr<StreamSwitchTaskInfo> &task_info) in StreamSwitchTask() argument
25 : TaskRepeater<StreamSwitchTaskInfo>(model_context, task_info), in StreamSwitchTask()
26 task_info_(task_info), in StreamSwitchTask()
29 MS_EXCEPTION_IF_NULL(task_info); in StreamSwitchTask()
33 } else if (stream_list_.size() > task_info->stream_id()) { in StreamSwitchTask()
34 stream_ = stream_list_[task_info->stream_id()]; in StreamSwitchTask()
36 …MS_LOG(EXCEPTION) << "Index: " << task_info->stream_id() << " >= stream_list.size(): " << stream_l… in StreamSwitchTask()
|
| D | task_factory.h | 51 …Task> Create(const ModelContext &model_context, const std::shared_ptr<TaskInfo> &task_info) const { in Create() argument
52 if (task_info == nullptr) { in Create()
57 auto iter = creator_map_.find(task_info->type()); in Create()
59 MS_LOG(ERROR) << "Unknown task type " << task_info->type(); in Create()
62 return iter->second(model_context, task_info); in Create()
78 …type, [](const ModelContext &model_context, const std::shared_ptr<TaskInfo> &task_info) -> std::sh…
79 …<task_info_clazz> concrete_task_info = std::static_pointer_cast<task_info_clazz>(task_info); \
|
| D | memcpy_async_task.cc | 23 const std::shared_ptr<MemcpyAsyncTaskInfo> &task_info) in MemcpyAsyncTask() argument
24 …: TaskRepeater<MemcpyAsyncTaskInfo>(model_context, task_info), task_info_(task_info), stream_(null… in MemcpyAsyncTask()
25 MS_EXCEPTION_IF_NULL(task_info); in MemcpyAsyncTask()
27 uint32_t stream_id = task_info->stream_id(); in MemcpyAsyncTask()
31 …MS_LOG(EXCEPTION) << "Index: " << task_info->stream_id() << " >= stream_list.size(): " << stream_l… in MemcpyAsyncTask()
|
| D | stream_active_task.cc | 23 const std::shared_ptr<StreamActiveTaskInfo> &task_info) in StreamActiveTask() argument
24 : TaskRepeater<StreamActiveTaskInfo>(model_context, task_info), in StreamActiveTask()
25 task_info_(task_info), in StreamActiveTask()
28 MS_EXCEPTION_IF_NULL(task_info); in StreamActiveTask()
30 uint32_t stream_id = task_info->stream_id(); in StreamActiveTask()
31 uint32_t active_stream_id = task_info->active_stream_id(); in StreamActiveTask()
|
| D | profiler_task.cc | 21 …lerTask(const ModelContext &model_context, const std::shared_ptr<ProfilerTraceTaskInfo> &task_info) in ProfilerTask() argument
22 …: TaskRepeater<ProfilerTraceTaskInfo>(model_context, task_info), task_info_(task_info), stream_(nu… in ProfilerTask()
23 MS_EXCEPTION_IF_NULL(task_info); in ProfilerTask()
25 uint32_t stream_id = task_info->stream_id(); in ProfilerTask()
28 …MS_LOG(EXCEPTION) << "Index: " << task_info->stream_id() << " >= stream_list.size(): " << stream_l… in ProfilerTask()
|
| D | tbe_task.cc | 24 TbeTask::TbeTask(const ModelContext &model_context, const std::shared_ptr<TbeTaskInfo> &task_info) in TbeTask() argument
25 : TaskRepeater<TbeTaskInfo>(model_context, task_info), in TbeTask()
26 task_info_(task_info), in TbeTask()
30 MS_EXCEPTION_IF_NULL(task_info); in TbeTask()
35 } else if (stream_list.size() > task_info->stream_id()) { in TbeTask()
36 stream_ = stream_list[task_info->stream_id()]; in TbeTask()
38 …MS_LOG(EXCEPTION) << "Index: " << task_info->stream_id() << " >= stream_list.size(): " << stream_l… in TbeTask()
|
| D | label_set_task.cc | 21 …LabelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInfo> &task_info) in LabelSetTask() argument
22 : TaskRepeater<LabelSetTaskInfo>(model_context, task_info), in LabelSetTask()
23 task_info_(task_info), in LabelSetTask()
29 uint32_t stream_id = task_info->stream_id(); in LabelSetTask()
30 uint32_t label_id = task_info->label_id(); in LabelSetTask()
|
| D | label_switch_task.cc | 22 const std::shared_ptr<LabelSwitchTaskInfo> &task_info) in LabelSwitchTask() argument
23 : TaskRepeater<LabelSwitchTaskInfo>(model_context, task_info), in LabelSwitchTask()
24 task_info_(task_info), in LabelSwitchTask()
27 MS_EXCEPTION_IF_NULL(task_info); in LabelSwitchTask()
32 uint32_t stream_id = task_info->stream_id(); in LabelSwitchTask()
|
| D | event_record_task.cc | 23 const std::shared_ptr<EventRecordTaskInfo> &task_info) in EventRecordTask() argument
24 : TaskRepeater<EventRecordTaskInfo>(model_context, task_info), in EventRecordTask()
25 task_info_(task_info), in EventRecordTask()
|
| D | event_wait_task.cc | 22 …entWaitTask(const ModelContext &model_context, const std::shared_ptr<EventWaitTaskInfo> &task_info) in EventWaitTask() argument
23 : TaskRepeater<EventWaitTaskInfo>(model_context, task_info), in EventWaitTask()
24 task_info_(task_info), in EventWaitTask()
|
| D | label_goto_task.cc | 22 …belGotoTask(const ModelContext &model_context, const std::shared_ptr<LabelGotoTaskInfo> &task_info) in LabelGotoTask() argument
23 : TaskRepeater<LabelGotoTaskInfo>(model_context, task_info), in LabelGotoTask()
24 task_info_(task_info), in LabelGotoTask()
|
| D | aicpu_task.cc | 26 …Task::AicpuTask(const ModelContext &model_context, const std::shared_ptr<AicpuTaskInfo> &task_info) in AicpuTask() argument
27 : TaskRepeater<AicpuTaskInfo>(model_context, task_info), in AicpuTask()
28 task_info_(task_info), in AicpuTask()
|
| D | hccl_task.cc | 28 …clTask::HcclTask(const ModelContext &model_context, const std::shared_ptr<HcclTaskInfo> &task_info) in HcclTask() argument
29 : TaskRepeater<HcclTaskInfo>(model_context, task_info), in HcclTask()
30 task_info_(task_info), in HcclTask()
|
| D | profiler_task.h | 26 …erTask(const ModelContext &model_context, const std::shared_ptr<ProfilerTraceTaskInfo> &task_info);
|
| D | memcpy_async_task.h | 26 …syncTask(const ModelContext &model_context, const std::shared_ptr<MemcpyAsyncTaskInfo> &task_info);
|
| D | label_set_task.h | 26 …abelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInfo> &task_info);
|
| D | stream_active_task.h | 26 …iveTask(const ModelContext &model_context, const std::shared_ptr<StreamActiveTaskInfo> &task_info);
|
| D | event_wait_task.h | 26 …ntWaitTask(const ModelContext &model_context, const std::shared_ptr<EventWaitTaskInfo> &task_info);
|
| D | event_record_task.h | 26 …cordTask(const ModelContext &model_context, const std::shared_ptr<EventRecordTaskInfo> &task_info);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | runtime_model.cc | 108 for (auto &task_info : davinci_model->GetTaskInfoList()) { in InitLabel() local
109 MS_EXCEPTION_IF_NULL(task_info); in InitLabel()
111 if (task_info->type() != TaskInfoType::LABEL_SET) { in InitLabel()
114 auto label_set_task_info = std::static_pointer_cast<LabelSetTaskInfo>(task_info); in InitLabel()
160 for (auto &task_info : task_infos) { in GenerateTask() local
161 auto task = TaskFactory::GetInstance().Create(model_context, task_info); in GenerateTask()
|
| /third_party/mesa3d/src/gallium/drivers/radeon/ |
| D | radeon_vce_40_2_2.c | 45 static void task_info(struct rvce_encoder *enc, uint32_t op, uint32_t dep, uint32_t fb_idx, in task_info() function
76 enc->task_info(enc, 0x00000000, 0, 0, 0); in create()
276 enc->task_info(enc, 0x00000002, 0, 0xffffffff, 0); in config()
291 enc->task_info(enc, 0x00000003, 0, 0, 0); in encode()
416 enc->task_info(enc, 0x00000001, 0, 0, 0); in destroy()
431 enc->task_info = task_info; in si_vce_40_2_2_init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_to_all.cc | 76 std::vector<hccl::HcclTaskInfo> task_info; in GenTask() local
77 bool ret = hccl::HcclAdapter::GetInstance().GenTask(anf_node, data_type_, &task_info); in GenTask()
83 for (auto &task : task_info) { in GenTask()
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | framework_parser.py | 455 for task_info in file:
456 infos = task_info.strip('\n').split(' ')
514 task_info = task_cache.get(result[0])
515 if task_info:
516 task_info.extend(result)
517 csv_writer.writerow(task_info)
|