Home
last modified time | relevance | path

Searched refs:MS_CTX_DEVICE_TARGET (Results 1 – 25 of 39) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dgroup_manager.cc78 (MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice)) { in CreateGroupByExecutor()
91 (MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice)) { in DestroyGroupByExecutor()
104 std::string device_name = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in CreateGroups()
112 (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice)) { in CreateGroups()
147 std::string device_name = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in CreateGroups()
197 std::string device_name = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in CreateGroup()
217 std::string device_name = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in DestroyGroup()
/third_party/mindspore/tests/ut/cpp/pynative/
Dpynative_execute_test.cc74 ASSERT_EQ(ctx3->get_param<std::string>(MS_CTX_DEVICE_TARGET), "CPU"); in TEST_F()
77 ctx3->set_param<std::string>(MS_CTX_DEVICE_TARGET, "GPU"); in TEST_F()
82 ASSERT_EQ(ctx4->get_param<std::string>(MS_CTX_DEVICE_TARGET), "GPU"); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Ddump_json_parser.cc190 ms_info["device_target"] = context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in CopyMSCfgJsonToDir()
242 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in ParseCommonDumpSetting()
244 } else if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice) { in ParseCommonDumpSetting()
289 if (e2e_dump_enabled_ && context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in ParseE2eDumpSetting()
370 } else if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kCPUDevice) { in ParseIteration()
495 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice) { in JudgeDumpEnabled()
499 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in JudgeDumpEnabled()
509 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) != kCPUDevice) { in JudgeDumpEnabled()
Ddump_utils.cc32 auto device_target = context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in ConvertPhysicalDeviceId()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc44 context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in TEST_F()
89 context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in TEST_F()
132 context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in TEST_F()
190 context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in TEST_F()
244 context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in TEST_F()
315 context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_optimization.cc202 is_gpu = (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice); in Run()
203 is_ascend = (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice); in Run()
/third_party/mindspore/mindspore/core/utils/
Dms_context.h118 MS_CTX_DEVICE_TARGET = MS_CTX_TYPE_STRING_BEGIN, enumerator
233 if (seter_ != nullptr && param == MS_CTX_DEVICE_TARGET) {
Dms_context.cc70 set_param<std::string>(MS_CTX_DEVICE_TARGET, target); in MsContext()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_input_to_attr.cc47 auto device = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in Process()
Dconst_to_attr_strided_slice_grad.cc126 if (ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in Process()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_split.cc32 std::string backend = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetWorldGroup()
Dpipeline.cc163 if (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in SetLoopCount()
165 } else if (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice) { in SetLoopCount()
1297 std::string device_name = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in InitHccl()
1300 ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in InitHccl()
1394 std::string device_name = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in StartUpProfiling()
1396 ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in StartUpProfiling()
/third_party/mindspore/mindspore/core/ir/
Danf.cc545 return context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetOriginNodeTarget()
551 std::string default_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetCNodeTarget()
562 std::string last_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in ContainMultiTarget()
/third_party/mindspore/mindspore/ccsrc/pybind_api/utils/
Dms_context_py.cc88 .value("device_target", MsCtxParam::MS_CTX_DEVICE_TARGET) in __anon7eb2a90d0202()
/third_party/mindspore/mindspore/core/ops/
Dbias_add.cc59 auto is_ascend = (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice); in InferShape()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime_manager.cc119 std::string device_name = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetCurrentKernelRuntime()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dbatchnorm_info.cc214 std::string backend = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in InferReplaceOps()
/third_party/mindspore/mindspore/ccsrc/utils/
Dcomm_manager.cc217 std::string backend = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetRank()
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/
Dps_cache_manager.h66 auto devcie_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in EmbeddingDeviceCache()
/third_party/mindspore/mindspore/ccsrc/utils/context/
Dgraph_kernel_flags.cc175 bool is_gpu = (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice); in RegisterFlags()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc104 context_ptr->set_param<std::string>(MS_CTX_DEVICE_TARGET, kDavinciMultiGraphInferenceDevice); in Build()
/third_party/mindspore/mindspore/ccsrc/cxx_api/graph/gpu/
Dgpu_graph_impl.cc58 ms_context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kGPUDevice); in InitEnv()
/third_party/mindspore/mindspore/ccsrc/vm/
Dtransform.cc573 std::string target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in CreateBackend()
612 std::string target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in SetMindRTEnable()
Dgraph_partition.cc43 std::string default_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetOtherTarget()
680 std::string default_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET); in Partition()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc262 if (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) != kAscendDevice && in IsNopNode()
263 context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) != kGPUDevice) { in IsNopNode()
722 auto device = ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET); in RectifyAbstractFromRegAttr()

12