Home
last modified time | relevance | path

Searched refs:kGPUDevice (Results 1 – 25 of 32) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_expander.cc67 {kGPUDevice, OpLevel_1, prim::kPrimBatchMatMul}, in GetExpandOps()
68 {kGPUDevice, OpLevel_0, prim::kPrimBiasAdd}, in GetExpandOps()
69 {kGPUDevice, OpLevel_1, prim::kPrimBiasAddGrad}, in GetExpandOps()
70 {kGPUDevice, OpLevel_0, prim::kPrimDropout}, in GetExpandOps()
71 {kGPUDevice, OpLevel_0, prim::kPrimDropoutGrad}, in GetExpandOps()
72 {kGPUDevice, OpLevel_0, prim::kPrimFusedAdam}, in GetExpandOps()
73 {kGPUDevice, OpLevel_0, prim::kPrimFusedAdamWeightDecay}, in GetExpandOps()
74 {kGPUDevice, OpLevel_1, prim::kPrimMaximumGrad}, in GetExpandOps()
75 {kGPUDevice, OpLevel_1, prim::kPrimMinimumGrad}, in GetExpandOps()
76 {kGPUDevice, OpLevel_1, prim::kPrimLayerNorm}, in GetExpandOps()
[all …]
Dgraph_kernel_cluster.cc62 {kGPUDevice, OpLevel_0, prim::kPrimACos}, in GetClusterableOpList()
63 {kGPUDevice, OpLevel_0, prim::kPrimAcosh}, in GetClusterableOpList()
64 {kGPUDevice, OpLevel_1, prim::kPrimArgMax}, in GetClusterableOpList()
65 {kGPUDevice, OpLevel_1, prim::kPrimArgMin}, in GetClusterableOpList()
66 {kGPUDevice, OpLevel_0, prim::kPrimAsin}, in GetClusterableOpList()
67 {kGPUDevice, OpLevel_0, prim::kPrimAsinh}, in GetClusterableOpList()
68 {kGPUDevice, OpLevel_0, prim::kPrimAssign}, in GetClusterableOpList()
69 {kGPUDevice, OpLevel_0, prim::kPrimAtan}, in GetClusterableOpList()
70 {kGPUDevice, OpLevel_0, prim::kPrimAtan2}, in GetClusterableOpList()
71 {kGPUDevice, OpLevel_0, prim::kPrimCos}, in GetClusterableOpList()
[all …]
Dparallel_cost_model.cc115 if (target != kGPUDevice) { in GetParallelCostModel()
116 MS_LOG(EXCEPTION) << "Parallel cost model only support " << kGPUDevice << " now."; in GetParallelCostModel()
Dgraph_kernel_optimization.cc180 pm->AddPass(std::make_shared<ParallelOpFusion>(kGPUDevice, ParallelConfig(7)), level, is_gpu); in Combine()
202 is_gpu = (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc133 InitExecutor(kGPUDevice, device_id); in Init()
146 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in StartKernelRT()
249 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in AllocateMemory()
257 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in RunOpAllocateMemory()
264 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in RunOpGenKernelEvent()
271 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in RunOpClearMemory()
404 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in CompileGraphImpl()
607 … device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in UpdateOutputTensors()
631 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in Execute()
722 …instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in DumpDataEnabledIteration()
[all …]
Dgpu_session.h104 MS_REG_SESSION(kGPUDevice, GPUSession);
Dexecutor.cc433 if (target == kGPUDevice) { in RunOp()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_input_to_attr.cc49 if (device != kGPUDevice) { in Process()
54 if (device == kGPUDevice) { in Process()
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger.cc212 if (device_target_ == kGPUDevice) { in CheckDebuggerDumpEnabled()
282 if (device_target_ != kGPUDevice) { in PreExecuteGraphDebugger()
320 if (device_target_ == kGPUDevice && num_step_ != 0) { in PreExecute()
331 …e if (debugger_enabled_ && graph_id == rungraph_id_list_.front() && device_target_ == kGPUDevice) { in PreExecute()
439 if (device_target_ != kGPUDevice) { in PostExecuteGraphDebugger()
472 if (device_target_ != kGPUDevice) { in PostExecute()
478 if (device_target_ != kGPUDevice) { in PostExecute()
484 if (device_target_ != kGPUDevice) { in PostExecute()
1320 …sContext::GetInstance()->get_param<bool>(MS_CTX_ENABLE_MINDRT) && (device_target_ == kGPUDevice)) { in LoadSingleAnfnode()
1437 …if (device_target_ == kGPUDevice && (debugger_enabled_ || device::KernelRuntime::DumpDataEnabledIt… in UpdateStepNum()
[all …]
/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()
112 (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice)) { in CreateGroups()
Dcostmodel_context.cc95 if (device_target == kGPUDevice) { in set_costmodel_context_for_device()
/third_party/mindspore/mindspore/core/utils/
Dms_context.h51 const char kGPUDevice[] = "GPU"; variable
60 const std::set<std::string> kTargetSet = {kCPUDevice, kGPUDevice, kAscendDevice, kDavinciDevice};
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/
Dgpu_ps_cache.cc28 MS_REG_PS_CACHE(kGPUDevice, GPUPsCache);
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_split.cc35 } else if (backend == kGPUDevice) { in GetWorldGroup()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dtransfer_node.cc88 } else if (device_type_ == kGPUDevice) { in Build()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.h140 MS_REG_KERNEL_RUNTIME(kGPUDevice, GPUKernelRuntime);
/third_party/mindspore/mindspore/ccsrc/utils/
Dcomm_manager.cc220 } else if (backend == kGPUDevice) { in GetRank()
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Ddump_json_parser.cc244 } else if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice) { in ParseCommonDumpSetting()
495 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice) { in JudgeDumpEnabled()
De2e_dump.cc42 return context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice; in IsDeviceTargetGPU()
/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()
Dcontext_extends.cc373 device_type_seter.reset(new (std::nothrow) MsContext("ms", kGPUDevice)); in DeviceTypeSetRegister()
/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/backend/kernel_compiler/oplib/
Doplib.cc359 bool is_gpu = (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kGPUDevice); in FindOp()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc263 context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) != kGPUDevice) { in IsNopNode()
723 if (device == kGPUDevice) { in RectifyAbstractFromRegAttr()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc536 MS_REGISTER_DEVICE(kGPUDevice, GPUDeviceContext);

12