Home
last modified time | relevance | path

Searched refs:kAscendDevice (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_expander.cc61 {kAscendDevice, OpLevel_0, prim::kLambApplyOptimizerAssign}, in GetExpandOps()
62 {kAscendDevice, OpLevel_0, prim::kLambApplyWeightAssign}, in GetExpandOps()
63 {kAscendDevice, OpLevel_0, prim::kPrimClipByNormNoDivSum}, in GetExpandOps()
64 {kAscendDevice, OpLevel_0, prim::kPrimSqrtGrad}, in GetExpandOps()
65 {kAscendDevice, OpLevel_1, prim::kSoftmaxGradExt}, in GetExpandOps()
66 {kAscendDevice, OpLevel_0, prim::kFusedMulAdd}, in GetExpandOps()
Dgraph_kernel_cluster.cc58 {kAscendDevice, OpLevel_1, prim::kPrimMatMul}, in GetClusterableOpList()
59 {kAscendDevice, OpLevel_1, prim::kPrimTransData}, in GetClusterableOpList()
60 {kAscendDevice, OpLevel_1, prim::kPrimBatchMatMul}, in GetClusterableOpList()
Dgraph_kernel_optimization.cc203 is_ascend = (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice); in Run()
/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/session/
Dascend_session.cc379 void AscendSession::Init(uint32_t device_id) { InitExecutor(kAscendDevice, device_id); } in Init()
514 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in LoadInputData()
629 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in CompileGraphImpl()
740 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in CompileChildGraph()
1229 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in InitRuntimeResource()
1333 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in BuildDynamicKernel()
1473 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in MemoryAlloc()
1482 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in RunOpMemoryAlloc()
1490 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in RunOpMemoryAllocNew()
1497 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in RunOpGenKernelEvent()
[all …]
Dsession_context.h34 explicit Context(std::string target = kAscendDevice, uint32_t device_id = 0)
Dexecutor.cc493 if (device_name_ == kAscendDevice) { in OnWorkerExit()
494 device::KernelRuntimeManager::Instance().ReleaseKernelRuntime(kAscendDevice, device_id_); in OnWorkerExit()
Dascend_session.h180 MS_REG_SESSION(kAscendDevice, AscendSession);
/third_party/mindspore/mindspore/core/utils/
Dms_context.h52 const char kAscendDevice[] = "Ascend"; variable
60 const std::set<std::string> kTargetSet = {kCPUDevice, kGPUDevice, kAscendDevice, kDavinciDevice};
Dms_context.cc82 set_param<bool>(MS_CTX_ENABLE_LOOP_SINK, target == kAscendDevice || target == kDavinciDevice); in MsContext()
/third_party/mindspore/mindspore/ccsrc/utils/context/
Dcontext_extends.cc369 device_type_seter.reset(new (std::nothrow) MsContext("ge", kAscendDevice)); in DeviceTypeSetRegister()
371 device_type_seter.reset(new (std::nothrow) MsContext("ms", kAscendDevice)); in DeviceTypeSetRegister()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
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.cc33 if (backend == kAscendDevice) { in GetWorldGroup()
Dpipeline.cc163 if (context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in SetLoopCount()
1300 ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in InitHccl()
1396 ms_context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in StartUpProfiling()
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Ddump_json_parser.cc242 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in ParseCommonDumpSetting()
289 if (e2e_dump_enabled_ && context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in ParseE2eDumpSetting()
499 if (context->get_param<std::string>(MS_CTX_DEVICE_TARGET) == kAscendDevice) { in JudgeDumpEnabled()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.h125 MS_REG_KERNEL_RUNTIME(kAscendDevice, AscendKernelRuntime);
Dascend_device_address.cc80 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in SyncMemory()
106 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in DataSync()
172 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in SyncStream()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dtransfer_node.cc90 } else if (device_type_ == kAscendDevice) { in Build()
/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/ascend/executor/
Dai_core_dynamic_kernel.cc155 …auto runtime_instance = KernelRuntimeManager::Instance().GetSingleKernelRuntime(kAscendDevice, dev… in AllocateWorkspace()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dbatchnorm_info.cc216 if (backend != kAscendDevice && backend != kDavinciDevice) { in InferReplaceOps()
/third_party/mindspore/mindspore/ccsrc/utils/
Dcomm_manager.cc218 if (backend == kAscendDevice) { in GetRank()
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/
Dascend_ps_cache.cc39 MS_REG_PS_CACHE(kAscendDevice, AscendPsCache);
/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc72 if (device == kAscendDevice) { in GetDependsFormMap()
/third_party/mindspore/mindspore/ccsrc/cxx_api/graph/ascend/
Dascend_graph_impl.cc306 ms_context->set_param<std::string>(MS_CTX_DEVICE_TARGET, kAscendDevice); in MsEnvGuard()

12