| /third_party/mindspore/mindspore/core/utils/ |
| D | ms_context.cc | 60 set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id); in MsContext()
63 set_param<uint32_t>(MS_CTX_DEVICE_ID, 0); in MsContext()
66 set_param<uint32_t>(MS_CTX_DEVICE_ID, 0); in MsContext()
126 uint32_t device_id = get_param<uint32_t>(MS_CTX_DEVICE_ID); in CreateTensorPrintThread()
|
| D | ms_context.h | 105 MS_CTX_DEVICE_ID = MS_CTX_TYPE_UINT32_BEGIN, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | group_manager.cc | 105 uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in CreateGroups()
148 uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in CreateGroups()
198 uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in CreateGroup()
218 uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in DestroyGroup()
|
| /third_party/mindspore/mindspore/ccsrc/utils/context/ |
| D | context_extends.cc | 69 uint32_t device_id = ms_context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in OpenTsd()
122 uint32_t device_id = ms_context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in CloseTsd()
244 auto env_device_id = std::to_string(ms_context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID)); in SetHcclOptions()
|
| /third_party/mindspore/mindspore/ccsrc/profiler/device/cpu/ |
| D | cpu_data_saver.cc | 37 auto device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in WriteFile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
| D | dataset_profiling.cc | 34 auto device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in GetDeviceId()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/ |
| D | profiling_utils.cc | 408 …TaskDescReporter task_reporter(context->get_param<uint32_t>(MS_CTX_DEVICE_ID), "vm_task_desc_info"… in ReportProfilingData()
413 …GraphDescReporter graph_reporter(context->get_param<uint32_t>(MS_CTX_DEVICE_ID), "vm_graph_desc_in… in ReportProfilingData()
423 PointReporter point_reporter(context->get_param<uint32_t>(MS_CTX_DEVICE_ID), "vm_point"); in ReportProfilingData()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 62 auto device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in KernelFusion()
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/utils/ |
| D | ms_context_py.cc | 95 .value("device_id", MsCtxParam::MS_CTX_DEVICE_ID) in __anon41857d0f0202()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime_manager.cc | 118 uint32_t device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in GetCurrentKernelRuntime()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.cc | 78 auto device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in SyncMemory()
105 auto device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in DataSync()
171 auto device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in SyncStream()
|
| D | ascend_memory_manager.cc | 63 unsigned int device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in MallocDeviceMemory()
|
| D | ascend_kernel_runtime.cc | 281 uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in ReleaseDeviceRes()
1111 …cclAdapter::GetInstance().InitHccl(context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID), rank_id_str, in HcclInit()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.cc | 154 auto device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in AllocateWorkspace()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/gpu/ |
| D | gpu_graph_impl.cc | 57 ms_context->set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id_); in InitEnv()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | graph_converter.cc | 242 const auto &device_id = context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/ascend/ |
| D | ascend_graph_impl.cc | 305 ms_context->set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id_); in MsEnvGuard()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 1278 uint32_t device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in InitHccl()
1293 ms_context->set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id); in InitHccl()
1393 uint32_t device_id = ms_context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in StartUpProfiling()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_utils.cc | 181 soc_info_json["deviceId"] = std::to_string(context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID)); in GenSocInfo()
|
| D | ascend_kernel_compile.cc | 485 static uint32_t source_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in JsonAssemble()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.cc | 79 ms_context->set_param<uint32_t>(MS_CTX_DEVICE_ID, device_context_key_.device_id_); in Initialize()
|
| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | dump_json_parser.cc | 510 auto device_id = context->get_param<uint32_t>(MS_CTX_DEVICE_ID); in JudgeDumpEnabled()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_creator.cc | 248 auto device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in GenFusionOpName()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | transform.cc | 574 uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); in CreateBackend()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.cc | 116 ms_context->set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id); in Init()
|