| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.h | 52 const DeviceAddress *GetOutputAddr(size_t index) const;
56 DeviceAddress *GetWorkspaceAddr(size_t index) const;
73 …const std::vector<std::shared_ptr<DeviceAddress>> &output_address_list() const { return output_add… in output_address_list()
74 …const std::vector<std::shared_ptr<DeviceAddress>> &workspace_address_list() const { return workspa… in workspace_address_list()
79 std::vector<std::shared_ptr<DeviceAddress>> output_address_list_;
80 std::vector<std::shared_ptr<DeviceAddress>> workspace_address_list_;
|
| D | device_address.h | 66 class DeviceAddress : public mindspore::DeviceSync {
68 explicit DeviceAddress(void *ptr, size_t size) : ptr_(ptr), size_(size) {} in DeviceAddress() function
69 explicit DeviceAddress(void *ptr, size_t size, const string &format, TypeId type_id) in DeviceAddress() function
71 explicit DeviceAddress(void *ptr, size_t size, const std::string &format, TypeId type_id, in DeviceAddress() function
74 virtual ~DeviceAddress() { ptr_ = nullptr; } in ~DeviceAddress()
133 using DeviceAddressPtr = std::shared_ptr<DeviceAddress>;
|
| D | kernel_info.cc | 25 const DeviceAddress *KernelInfo::GetOutputAddr(size_t index) const { in GetOutputAddr()
68 DeviceAddress *KernelInfo::GetWorkspaceAddr(size_t index) const { in GetWorkspaceAddr()
|
| D | bucket.cc | 76 auto device_address = std::dynamic_pointer_cast<DeviceAddress>(tensor->device_address()); in UpdateTensorAddr()
|
| D | kernel_runtime.cc | 256 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpMallocPre()
407 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpAssignInputMemory()
450 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(device_address), iter.seco… in RunOpAssignOutputMemory()
540 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(pre_output_tensors[i]->dev… in RunOpAssignOutputNodeMemory()
543 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(pre_output_tensors[i]->dev… in RunOpAssignOutputNodeMemory()
936 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignExtraStaticMem()
965 … auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignValueNodeTensor()
967 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()),… in AssignValueNodeTensor()
1287 … const DeviceAddress *device_address, const kernel::AddressPtr &kernel_addr) { in GetOrMallocAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_device_address.h | 27 class CPUDeviceAddress : public DeviceAddress {
29 CPUDeviceAddress(void *ptr, size_t size) : DeviceAddress(ptr, size) {} in CPUDeviceAddress()
32 : DeviceAddress(ptr, size, format, type_id) {} in CPUDeviceAddress()
35 : DeviceAddress(ptr, size, format, type_id, node_index) {} in CPUDeviceAddress()
|
| D | cpu_kernel_runtime.h | 69 void AddRuntimeAddress(DeviceAddress *address, std::vector<kernel::AddressPtr> *input_list);
|
| D | cpu_kernel_runtime.cc | 102 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()),… in AssignValueNodeAddress()
374 auto address_ptr = std::dynamic_pointer_cast<device::DeviceAddress>(address); in BindOutputTensorAddressPtr()
391 void CPUKernelRuntime::AddRuntimeAddress(DeviceAddress *address, std::vector<kernel::AddressPtr> *i… in AddRuntimeAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_address.h | 32 class GPUDeviceAddress : public DeviceAddress {
34 GPUDeviceAddress(void *ptr, size_t size) : DeviceAddress(ptr, size) {} in GPUDeviceAddress()
36 : DeviceAddress(ptr, size, format, type_id) {} in GPUDeviceAddress()
38 : DeviceAddress(ptr, size, format, type_id, node_index) {} in GPUDeviceAddress()
|
| D | gpu_bucket.cc | 59 …DeviceAddressPtr device_address = std::dynamic_pointer_cast<DeviceAddress>(tensor->device_address(… in AllocateAllReduceAddr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.h | 36 class AscendDeviceAddress : public DeviceAddress {
38 explicit AscendDeviceAddress(void *ptr, size_t size) : DeviceAddress(ptr, size) {} in AscendDeviceAddress()
40 : DeviceAddress(ptr, size, format, type_id) {} in AscendDeviceAddress()
43 : DeviceAddress(ptr, size, format, type_id, node_index) {} in AscendDeviceAddress()
|
| D | ascend_bucket.cc | 44 …DeviceAddressPtr device_address = std::dynamic_pointer_cast<DeviceAddress>(tensor->device_address(… in AllocateAllReduceAddr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.h | 40 bool AllocateMemory(DeviceAddress *const &address, size_t size) const override;
41 void FreeMemory(DeviceAddress *const &address) const override;
|
| D | cpu_device_context.cc | 78 bool CPUDeviceContext::AllocateMemory(DeviceAddress *const &address, size_t size) const { in AllocateMemory()
91 void CPUDeviceContext::FreeMemory(DeviceAddress *const &address) const { in FreeMemory()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/ |
| D | device_context.h | 59 virtual bool AllocateMemory(DeviceAddress *const &address, size_t size) const = 0;
60 virtual void FreeMemory(DeviceAddress *const &address) const = 0;
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.h | 42 bool AllocateMemory(DeviceAddress *const &address, size_t size) const override;
43 void FreeMemory(DeviceAddress *const &address) const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 44 using DeviceAddress = device::DeviceAddress; variable
187 …static const DeviceAddress *GetOutputAddr(const AnfNodePtr &node, size_t output_idx, bool visit_no…
195 static const DeviceAddress *GetPrevNodeOutputAddr(const AnfNodePtr &node, size_t input_idx,
204 static DeviceAddress *GetWorkspaceAddr(const AnfNodePtr &node, size_t output_idx);
|
| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | dump_utils.h | 38 void DumpMemToFile(const std::string &file_path, const device::DeviceAddress &addr, const ShapeVect…
|
| D | e2e_dump.h | 73 const device::DeviceAddress &addr, const ShapeVector &int_shapes,
|
| D | dump_utils.cc | 124 void DumpMemToFile(const std::string &file_path, const device::DeviceAddress &addr, const ShapeVect… in DumpMemToFile()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | device_tensor_store.h | 28 using DeviceTensor = mindspore::device::DeviceAddress;
|
| /third_party/vk-gl-cts/external/vulkancts/framework/vulkan/ |
| D | vkRayTracingUtil.cpp | 563 … MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress)); in SerialStorage()
568 … MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress)); in SerialStorage()
586 … MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress)); in SerialStorage()
794 …, MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress); in createVertexBuffer()
835 …, MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress); in createIndexBuffer()
1092 … MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress)); in create()
1097 … MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress)); in create()
1125 … MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress)); in create()
1592 …| MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress); in createInstanceBuffer()
1597 …, MemoryRequirement::HostVisible | MemoryRequirement::Coherent | MemoryRequirement::DeviceAddress); in createInstanceBuffer()
[all …]
|
| D | vkMemUtil.cpp | 127 const MemoryRequirement MemoryRequirement::DeviceAddress = MemoryRequirement(MemoryRequirement::FLA… member in vk::MemoryRequirement
238 if (requirement & MemoryRequirement::DeviceAddress) in allocate()
|
| D | vkMemUtil.hpp | 92 static const MemoryRequirement DeviceAddress; member in vk::MemoryRequirement
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | actor_common.h | 37 using DeviceTensor = mindspore::device::DeviceAddress;
|