Home
last modified time | relevance | path

Searched refs:device_context (Results 1 – 25 of 41) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dmemory_manager_actor.cc26 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context, in AllocateMemory() argument
29 MS_EXCEPTION_IF_NULL(device_context); in AllocateMemory()
38 if (!device_context->AllocateMemory(device_tensor, device_tensor->GetSize())) { in AllocateMemory()
39 …SetOpContextMemoryAllocFail(from_aid.Name(), device_context, device_tensor->GetSize(), op_context); in AllocateMemory()
69 auto &device_context = (*device_contexts)[i]; in AllocateContinuousMemory() local
70 MS_EXCEPTION_IF_NULL(device_context); in AllocateContinuousMemory()
72 if (!device_context->AllocateContinuousMemory(alloc_list, total_size, size_list)) { in AllocateContinuousMemory()
73 SetOpContextMemoryAllocFail(from_aid.Name(), device_context, total_size, op_context); in AllocateContinuousMemory()
95 auto &device_context = (*device_contexts)[i]; in AllocateBatchMemory() local
97 MS_EXCEPTION_IF_NULL(device_context); in AllocateBatchMemory()
[all …]
Ddata_prepare_actor.cc32 const DeviceContext *device_context, OpContext<DeviceTensor> *const context, in SyncTensorData() argument
37 MS_EXCEPTION_IF_NULL(device_context); in SyncTensorData()
41 (!device_context->AllocateMemory(device_tensor.get(), device_tensor->GetSize()))) { in SyncTensorData()
42 …SET_OPCONTEXT_MEMORY_ALLOC_FAIL_BY_STRATEGY(strategy, *context, *device_context, node->fullname_wi… in SyncTensorData()
197 const auto &device_context = graph_compiler_info_->device_contexts_[i]; in PrepareDataForDeviceTensorStore() local
202 PrepareDataForValueNode(value_node, device_context, context); in PrepareDataForDeviceTensorStore()
217 PrepareDataForWeightNode(input_node, front_node, input_tensor, device_context, context); in PrepareDataForDeviceTensorStore()
278 const auto &device_context = graph_compiler_info_->device_contexts_[i]; in PrepareDataForStepMode() local
280 MS_EXCEPTION_IF_NULL(device_context); in PrepareDataForStepMode()
314 auto device_address = device_context->CreateDeviceAddress( in PrepareDataForStepMode()
[all …]
Ddebug_actor.cc35 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context, in Debug() argument
38 MS_EXCEPTION_IF_NULL(device_context); in Debug()
49 if (device_context->GetDeviceAddressType() == device::DeviceAddressType::kCPU) { in Debug()
57 } else if (device_context->GetDeviceAddressType() == device::DeviceAddressType::kGPU) { in Debug()
Dmemory_manager_actor.h40 … AllocateMemory(const std::vector<DeviceTensor *> *alloc_list, const DeviceContext *device_context,
55 void FreeMemory(const std::vector<DeviceTensor *> *free_list, const DeviceContext *device_context,
67 …id SetOpContextMemoryAllocFail(const std::string &kernel_name, const DeviceContext *device_context,
Ddata_prepare_actor.h80 … const DeviceContext *device_context, OpContext<DeviceTensor> *const context);
82 void PrepareDataForValueNode(const ValueNodePtr &node, const DeviceContext *device_context,
86 … const DeviceContext *device_context, OpContext<DeviceTensor> *const context);
97 … &node, const AnfNodePtr &front_node, const TensorPtr &tensor, const DeviceContext *device_context,
Drecorder_actor.cc29 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context) { in RecordInfo() argument
31 MS_EXCEPTION_IF_NULL(device_context); in RecordInfo()
Doutput_actor.cc93 const auto &device_context = device_contexts_[i]; in UpdateOutputDeviceAddress() local
94 MS_EXCEPTION_IF_NULL(device_context); in UpdateOutputDeviceAddress()
95 … auto new_device_tensor = device_context->CreateDeviceAddress(nullptr, device_tensor->GetSize(), in UpdateOutputDeviceAddress()
Dkernel_actor.cc152 … AllocateMemory(const std::vector<DeviceTensor *> &alloc_list, const DeviceContext *device_context, in AllocateMemory() argument
154 MS_EXCEPTION_IF_NULL(device_context); in AllocateMemory()
163 if (!device_context->AllocateMemory(device_tensor, device_tensor->GetSize())) { in AllocateMemory()
164 …MEMORY_ALLOC_FAIL_BY_STRATEGY(GraphExecutionStrategy::kStep, *context, *device_context, actor_name, in AllocateMemory()
170 void FreeMemory(const std::vector<DeviceTensor *> &free_list, const DeviceContext *device_context) { in FreeMemory() argument
171 MS_EXCEPTION_IF_NULL(device_context); in FreeMemory()
182 device_context->FreeMemory(device_tensor); in FreeMemory()
Dgather_actor.cc152 const auto &device_context = device_contexts_[device_tensor_store_key.first]; in FetchInputDeviceTensor() local
153 MS_EXCEPTION_IF_NULL(device_context); in FetchInputDeviceTensor()
155 …DeviceTensorStore::GetInstance().Fetch(device_tensor_store_key.second, device_context->GetDeviceAd… in FetchInputDeviceTensor()
159 ", device type:" + std::to_string(static_cast<int>(device_context->GetDeviceAddressType())); in FetchInputDeviceTensor()
Dactor_common.h85 #define SET_OPCONTEXT_MEMORY_ALLOC_FAIL_BY_STRATEGY(strategy, op_context, device_context, kernel_na… argument
87 …std::string message = "Device(id:" + std::to_string((device_context).device_context_key().device_i…
Dkernel_actor.h46 KernelActor(const std::string &name, const CNodePtr &kernel, const DeviceContext *device_context, in KernelActor() argument
55 (void)device_contexts_.emplace_back(device_context); in KernelActor()
Dswitch_actor.h62 …SwitchActor(const std::string &name, DeviceContext *device_context, const CNodePtr &node, const in… in SwitchActor() argument
65 device_context_(device_context), in SwitchActor()
Ddata_source_actor.h86 …taSourceActor(const std::string &name, size_t buffer_capacity, const DeviceContext *device_context, in DeviceQueueDataSourceActor() argument
90 (void)device_contexts_.emplace_back(device_context); in DeviceQueueDataSourceActor()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc46 bool NodeDeviceAddressExist(const DeviceContext *device_context, const AnfNodePtr &kernel, size_t i… in NodeDeviceAddressExist() argument
48 MS_EXCEPTION_IF_NULL(device_context); in NodeDeviceAddressExist()
52 return address->DeviceType() == device_context->GetDeviceAddressType(); in NodeDeviceAddressExist()
57 void CreateParameterDeviceAddress(const DeviceContext *device_context, const KernelGraphPtr &graph)… in CreateParameterDeviceAddress() argument
58 MS_EXCEPTION_IF_NULL(device_context); in CreateParameterDeviceAddress()
77 if (!out->isa<Parameter>() || NodeDeviceAddressExist(device_context, out, 0)) { in CreateParameterDeviceAddress()
83 if (!item->isa<Parameter>() || NodeDeviceAddressExist(device_context, item, 0)) { in CreateParameterDeviceAddress()
99 auto device_address = device_context->CreateDeviceAddress(nullptr, tensor_size, in CreateParameterDeviceAddress()
106 void CreateDeviceAddressForTensorValue(const DeviceContext *device_context, const ValuePtr &node_va… in CreateDeviceAddressForTensorValue() argument
108 MS_EXCEPTION_IF_NULL(device_context); in CreateDeviceAddressForTensorValue()
[all …]
Dcontrol_node_parser.cc141 const DeviceContext *device_context) { in CreateDeviceTensorForValueNode() argument
142 MS_EXCEPTION_IF_NULL(device_context); in CreateDeviceTensorForValueNode()
169 device_context->CreateDeviceAddress(nullptr, tensor_size, output_format, output_type_id); in CreateDeviceTensorForValueNode()
177 … CreateDeviceTensorForFrontParameter(const AnfNodePtr &node, const DeviceContext *device_context) { in CreateDeviceTensorForFrontParameter() argument
178 MS_EXCEPTION_IF_NULL(device_context); in CreateDeviceTensorForFrontParameter()
193 …device::DeviceAddressPtr address = device_context->CreateDeviceAddress(nullptr, size, kOpFormat_DE… in CreateDeviceTensorForFrontParameter()
927 …const auto &device_context = front_to_backend_parameters_[parameters[i - kCallInputStartPos]].seco… in FetchFrontValueNode() local
928 CreateDeviceTensorForValueNode(inputs[i], backend_node, device_context); in FetchFrontValueNode()
929 (void)front_value_nodes_.emplace_back(inputs[i], device_context); in FetchFrontValueNode()
968 const auto &device_context = call_node_to_backend_parameter.second.second; in FetchFrontValueNode() local
[all …]
Dgraph_compiler.h100 …h(const AnfNodePtrList &nodes, const AnfNodePtrList &outputs, const DeviceContext *device_context);
105 bool *single_op_cache_hit, const DeviceContext *device_context);
175 GraphId CompileGraphImpl(const KernelGraphPtr &graph, const DeviceContext *device_context) const;
178 void CreateDeviceAddress(const KernelGraphPtr &graph, const DeviceContext *device_context) const;
/third_party/mindspore/mindspore/lite/src/common/
Dcontext_util.cc28 void PassBasicProperties(std::shared_ptr<T> device_info, const lite::DeviceContext &device_context)… in PassBasicProperties() argument
30 device_info->SetProvider(device_context.provider_); in PassBasicProperties()
31 device_info->SetProviderDevice(device_context.provider_device_); in PassBasicProperties()
32 device_info->SetAllocator(device_context.allocator_); in PassBasicProperties()
118 for (auto &device_context : context->device_list_) { in MSContextFromContext() local
119 auto device_type = device_context.device_type_; in MSContextFromContext()
124 auto device_info = transfer_funcs[device_type](device_context); in MSContextFromContext()
130 ms_context->SetThreadAffinity(device_context.device_info_.cpu_device_info_.cpu_bind_mode_); in MSContextFromContext()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/
Ddevice_context_manager.cc44 std::shared_ptr<DeviceContext> device_context; in GetOrCreateDeviceContext() local
47 device_context = (creator_iter->second)(device_context_key); in GetOrCreateDeviceContext()
48 MS_EXCEPTION_IF_NULL(device_context); in GetOrCreateDeviceContext()
49 device_contexts_[device_context_key_str] = device_context; in GetOrCreateDeviceContext()
54 return device_context.get(); in GetOrCreateDeviceContext()
/third_party/ffmpeg/libavutil/
Dhwcontext_d3d11va.c403 ID3D11DeviceContext_CopySubresourceRegion(device_hwctx->device_context, in d3d11va_transfer_data()
407 hr = ID3D11DeviceContext_Map(device_hwctx->device_context, in d3d11va_transfer_data()
417 ID3D11DeviceContext_Unmap(device_hwctx->device_context, staging, 0); in d3d11va_transfer_data()
419 hr = ID3D11DeviceContext_Map(device_hwctx->device_context, in d3d11va_transfer_data()
429 ID3D11DeviceContext_Unmap(device_hwctx->device_context, staging, 0); in d3d11va_transfer_data()
431 ID3D11DeviceContext_CopySubresourceRegion(device_hwctx->device_context, in d3d11va_transfer_data()
460 if (!device_hwctx->device_context) { in d3d11va_device_init()
461 ID3D11Device_GetImmediateContext(device_hwctx->device, &device_hwctx->device_context); in d3d11va_device_init()
462 if (!device_hwctx->device_context) in d3d11va_device_init()
474 … hr = ID3D11DeviceContext_QueryInterface(device_hwctx->device_context, &IID_ID3D11VideoContext, in d3d11va_device_init()
[all …]
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc237 const DeviceContext *device_context) { in UpdateOutputDeviceAddress() argument
251 MS_EXCEPTION_IF_NULL(device_context); in UpdateOutputDeviceAddress()
252 … auto new_device_tensor = device_context->CreateDeviceAddress(nullptr, device_tensor->GetSize(), in UpdateOutputDeviceAddress()
358 const auto &device_context = in MindRTBackend() local
360 device_context->Initialize(); in MindRTBackend()
361 device_id_ = device_context->device_context_key().device_id_; in MindRTBackend()
435 const auto &device_context = in CompileGraph() local
437 MS_EXCEPTION_IF_NULL(device_context); in CompileGraph()
438 device_context->Initialize(); in CompileGraph()
453 auto graph_id = graph_compiler_->CompileGraph(segment->nodes_, outputs, device_context); in CompileGraph()
[all …]
/third_party/gstreamer/gstplugins_bad/gst-libs/gst/d3d11/
Dgstd3d11device.cpp110 ID3D11DeviceContext *device_context; member
787 GST_D3D11_CLEAR_COM (priv->device_context); in gst_d3d11_device_dispose()
1031 ComPtr < ID3D11DeviceContext > device_context; in gst_d3d11_device_new_internal() local
1084 device->GetImmediateContext (&device_context); in gst_d3d11_device_new_internal()
1097 D3D11_SDK_VERSION, &device, &selected_level, &device_context); in gst_d3d11_device_new_internal()
1104 &selected_level, &device_context); in gst_d3d11_device_new_internal()
1114 D3D11_SDK_VERSION, &device, &selected_level, &device_context); in gst_d3d11_device_new_internal()
1121 &selected_level, &device_context); in gst_d3d11_device_new_internal()
1159 priv->device_context = device_context.Detach (); in gst_d3d11_device_new_internal()
1285 return device->priv->device_context; in gst_d3d11_device_get_device_context_handle()
[all …]
Dgstd3d11memory.cpp326 ID3D11DeviceContext *device_context = in gst_d3d11_memory_map_cpu_access() local
329 hr = device_context->Map (staging, 0, map_type, 0, &priv->map); in gst_d3d11_memory_map_cpu_access()
345 ID3D11DeviceContext *device_context; in gst_d3d11_memory_upload() local
351 device_context = gst_d3d11_device_get_device_context_handle (dmem->device); in gst_d3d11_memory_upload()
352 device_context->CopySubresourceRegion (priv->texture, priv->subresource_index, in gst_d3d11_memory_upload()
361 ID3D11DeviceContext *device_context; in gst_d3d11_memory_download() local
367 device_context = gst_d3d11_device_get_device_context_handle (dmem->device); in gst_d3d11_memory_download()
368 device_context->CopySubresourceRegion (priv->staging, 0, 0, 0, 0, in gst_d3d11_memory_download()
442 ID3D11DeviceContext *device_context = in gst_d3d11_memory_unmap_cpu_access() local
445 device_context->Unmap (staging, 0); in gst_d3d11_memory_unmap_cpu_access()
[all …]
/third_party/gstreamer/gstplugins_bad/sys/d3d11/
Dgstd3d11screencapture.cpp989 ID3D11DeviceContext *device_context = in CopyMove() local
1025 device_context->CopySubresourceRegion(move_texture_.Get(), in CopyMove()
1029 device_context->CopySubresourceRegion(SharedSurf, in CopyMove()
1179 ID3D11DeviceContext *device_context = in CopyDirty() local
1212 device_context->OMSetBlendState(nullptr, nullptr, 0xFFFFFFFF); in CopyDirty()
1213 device_context->OMSetRenderTargets(1, &rtv, nullptr); in CopyDirty()
1214 device_context->VSSetShader(vs_.Get(), nullptr, 0); in CopyDirty()
1215 device_context->PSSetShader(ps_.Get(), nullptr, 0); in CopyDirty()
1216 device_context->PSSetShaderResources(0, 1, &srv); in CopyDirty()
1217 device_context->PSSetSamplers(0, 1, &samplers); in CopyDirty()
[all …]
/third_party/flutter/skia/third_party/externals/imgui/examples/
Dimgui_impl_dx11.h16 …I_IMPL_API bool ImGui_ImplDX11_Init(ID3D11Device* device, ID3D11DeviceContext* device_context);
/third_party/skia/third_party/externals/imgui/backends/
Dimgui_impl_dx11.h19 …I_IMPL_API bool ImGui_ImplDX11_Init(ID3D11Device* device, ID3D11DeviceContext* device_context);

12