Searched refs:backend_policy (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/pynative/ |
| D | pynative_execute_test.cc | 73 ASSERT_EQ(ctx3->backend_policy(), "vm"); in TEST_F()
81 ASSERT_EQ(ctx4->backend_policy(), "ge_only"); in TEST_F()
88 ASSERT_EQ(std::string(ctx->backend_policy()), "ge_only"); in TEST_F()
|
| /third_party/mindspore/ |
| D | setup.py | 26 backend_policy = os.getenv('BACKEND_POLICY') variable
50 file.write("__backend__ = '{}'\n".format(backend_policy))
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | ms_context.h | 143 MsContext(const std::string &backend_policy, const std::string &target);
152 std::string backend_policy() const;
|
| D | ms_context.cc | 176 std::string MsContext::backend_policy() const { in backend_policy() function in mindspore::MsContext
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 665 std::string backend = ms_context->backend_policy(); in GetGeBackendPolicy()
690 std::string backend = MsContext::GetInstance()->backend_policy(); in GetPipeline()
961 if (ms_context->backend_policy() == kMsConvert && py::isinstance<py::array>(arg)) { in ProcessVmArgInner()
1030 std::string backend = MsContext::GetInstance()->backend_policy(); in Run()
1040 if (MsContext::GetInstance()->backend_policy() == kMsConvert) { in Run()
1150 std::string name = MsContext::GetInstance()->backend_policy(); in InitExecDataset()
1167 std::string backend = MsContext::GetInstance()->backend_policy(); in InitExecDataset()
1299 if (ms_context->backend_policy() == "ms" && in InitHccl()
1395 if (ms_context->backend_policy() == "ms" && in StartUpProfiling()
|
| D | action.cc | 707 std::string backend = MsContext::GetInstance()->backend_policy(); in TaskEmitAction()
761 std::string backend = MsContext::GetInstance()->backend_policy(); in ExecuteAction()
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/utils/ |
| D | ms_context_py.cc | 108 … .def("get_backend_policy", &mindspore::MsContext::backend_policy, "Get backend policy.") in __anon41857d0f0202()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 1801 auto backend_policy = InitEnv(op_exec_info); in RunOpWithInitBackendPolicy() local
1804 py::object result = RunOpWithBackendPolicy(backend_policy, op_exec_info, &status); in RunOpWithInitBackendPolicy()
1816 MsBackendPolicy backend_policy; in InitEnv() local
1825 if (ms_context->backend_policy() == "ms") { in InitEnv()
1826 backend_policy = kMsBackendMsPrior; in InitEnv()
1828 backend_policy = kMsBackendVmOnly; in InitEnv()
1834 backend_policy = kMsBackendGeOnly; in InitEnv()
1837 backend_policy = kMsBackendVmOnly; in InitEnv()
1839 return backend_policy; in InitEnv()
1842 py::object ForwardExecutor::RunOpWithBackendPolicy(MsBackendPolicy backend_policy, const OpExecInfo… in RunOpWithBackendPolicy() argument
[all …]
|
| D | pynative_execute.h | 337 …py::object RunOpWithBackendPolicy(MsBackendPolicy backend_policy, const OpExecInfoPtr &op_exec_inf…
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | transform.cc | 566 std::string name = context_ptr->backend_policy(); in CreateBackend()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | parse.cc | 1154 if (MsContext::GetInstance()->backend_policy() != "ge") { in ParseIf()
|