| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | multi_stream_controller.cc | 37 void MultiStreamController::Refresh(const DeviceContext *device_context) { in Refresh()
55 bool MultiStreamController::UpdateTaskIdOnStream(const DeviceContext *device_context, int64_t task_… in UpdateTaskIdOnStream()
60 int64_t MultiStreamController::QueryTaskIdOnStream(const DeviceContext *device_context, uint32_t us… in QueryTaskIdOnStream()
65 int64_t MultiStreamController::LaunchTaskIdOnStream(const DeviceContext *device_context, uint32_t s… in LaunchTaskIdOnStream()
81 int64_t MultiStreamController::GetTaskIdOnStream(const DeviceContext *device_context, uint32_t stre… in GetTaskIdOnStream()
85 std::mutex &MultiStreamController::GetStreamMutex(const DeviceContext *device_context, size_t strea… in GetStreamMutex()
89 bool MultiStreamController::RecordEvent(const DeviceContext *device_context, int64_t task_id_on_str… in RecordEvent()
107 bool MultiStreamController::WaitEvent(const DeviceContext *device_context, int64_t task_id_on_strea… in WaitEvent()
122 bool MultiStreamController::WaitEvent(const DeviceContext *device_context, int64_t task_id_on_strea… in WaitEvent()
133 bool MultiStreamController::DispatchRecordWaitEvent(const DeviceContext *device_context, uint32_t u… in DispatchRecordWaitEvent()
[all …]
|
| D | loadable_device_address.cc | 39 auto device_context = GetDeviceContext(); in Offload() local
61 auto device_context = GetDeviceContext(); in Load() local
106 const auto device_context = GetDeviceContext(); in MoveToHost() local
158 const auto device_context = GetDeviceContext(); in MoveToDevice() local
216 const auto device_context = GetDeviceContext(); in MoveToFile() local
273 const auto device_context = GetDeviceContext(); in CopyHostToFile() local
292 const auto device_context = GetDeviceContext(); in CopyFileToHost() local
319 auto device_context = GetDeviceContext(); in ReleaseResource() local
409 const auto device_context = GetDeviceContext(); in Wait() local
|
| D | device_address_utils.cc | 47 device::DeviceAddressPtr CreateDeviceAddressForScalarAndString(const DeviceContext *device_context, in CreateDeviceAddressForScalarAndString()
89 Format GetFormatByTensorShape(const DeviceContext *device_context, const ShapeVector &tensor_shape)… in GetFormatByTensorShape()
105 bool DeviceAddressUtils::NodeDeviceAddressExist(const DeviceContext *device_context, const AnfNodeP… in NodeDeviceAddressExist()
118 void DeviceAddressUtils::CopyNoneTensorDataToDevice(const device::DeviceContext *device_context, in CopyNoneTensorDataToDevice()
154 void DeviceAddressUtils::CreateDeviceAddressByMapTensorNode(const DeviceContext *device_context, co… in CreateDeviceAddressByMapTensorNode()
192 void DeviceAddressUtils::CreateParameterDeviceAddress(const DeviceContext *device_context, in CreateParameterDeviceAddress()
282 …sPtrList DeviceAddressUtils::CreateDeviceAddressForTensorValue(const DeviceContext *device_context, in CreateDeviceAddressForTensorValue()
380 device::DeviceAddressPtr CreateDeviceAddressForTypeValue(const DeviceContext *device_context, in CreateDeviceAddressForTypeValue()
396 void DeviceAddressUtils::CreateValueNodeDeviceAddress(const DeviceContext *device_context, in CreateValueNodeDeviceAddress()
463 void DeviceAddressUtils::CreateKernelOutputDeviceAddress(const DeviceContext *device_context, in CreateKernelOutputDeviceAddress()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/ |
| D | pyboost_utils.h | 65 …static void PrepareOpInputs(const DeviceContext *device_context, size_t stream_id, const Args &...… in PrepareOpInputs()
72 static void MallocOpInputs(const DeviceContext *device_context, const T &... args) { in MallocOpInputs()
79 static void MallocOpInputsForView(const DeviceContext *device_context, const T &... args) { in MallocOpInputsForView()
86 static void GetAddressInfoHelper(const DeviceContext *device_context, size_t stream_id, in GetAddressInfoHelper()
96 static AddressInfoPair GetAddressInfo(const DeviceContext *device_context, size_t stream_id, in GetAddressInfo()
116 static void GetKernelTensor(const DeviceContext *device_context, size_t stream_id, size_t index, in GetKernelTensor()
128 static void GetKernelTensor(const DeviceContext *device_context, size_t stream_id, in GetKernelTensor()
158 static void GetKernelTensor(const DeviceContext *device_context, size_t stream_id, in GetKernelTensor()
173 static void PrepareOpOutputs(const DeviceContext *device_context, size_t stream_id, in PrepareOpOutputs()
179 …static void MallocOpOutputs(const DeviceContext *device_context, const std::vector<tensor::BaseTen… in MallocOpOutputs()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/ |
| D | statistic_kernel.h | 45 …StatisticKernel(const DeviceContext *device_context, string kernel_name, const std::set<TypeId> &d… in StatisticKernel()
71 …explicit DimStatisticKernel(const DeviceContext *device_context, string kernel_name, const std::se… in DimStatisticKernel()
82 …explicit MeanStatisticKernel(const DeviceContext *device_context, const std::set<TypeId> &dtype_id) in MeanStatisticKernel()
88 …explicit NormStatisticKernel(const DeviceContext *device_context, const std::set<TypeId> &dtype_id) in NormStatisticKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 60 const device::DeviceContext *device_context) { in UpdateInputTensorFromDevice()
104 const device::DeviceContext *device_context, bool is_sync) { in SetDeviceAddress()
133 const device::DeviceContext *device_context, bool is_sync) { in UpdateInputNodeDeviceAddress()
159 const device::DeviceContext *device_context) { in CopyTensorDataToDevice()
189 …CopyValueNodeDataToDevice(const KernelGraphPtr &graph, const device::DeviceContext *device_context… in CopyValueNodeDataToDevice()
228 const device::DeviceContext *device_context) { in CopyMapTensorDataToDevice()
249 const device::DeviceContext *device_context) { in CopyParameterDataToDevice()
274 const device::DeviceContext *device_context, const CNodePtr &node) { in MallocForKernelInput()
306 const device::DeviceContext *device_context) { in MallocForKernelOutput()
379 … const device::DeviceContext *device_context, in GetWorkspaceKernelTensors()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | memory_manager_actor.cc | 35 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context, in AllocateMemory()
98 auto &device_context = (*device_contexts)[i]; in AllocateContinuousMemory() local
170 auto &device_context = (*device_contexts)[i]; in AllocateBatchMemory() local
200 …ManagerActor::AllocateSomasMemory(SomasInfo *const somas_info, const DeviceContext *device_context, in AllocateSomasMemory()
270 …ctor::FreeMemory(const std::vector<DeviceTensor *> *free_list, const DeviceContext *device_context, in FreeMemory()
293 auto &device_context = (*device_contexts)[i]; in FreeBatchMemory() local
300 …moryManagerActor::FreeSomasMemory(SomasInfo *const somas_info, const DeviceContext *device_context, in FreeSomasMemory()
364 …Actor::FreeMemoryByRefCount(DeviceTensor *const device_tensor, const DeviceContext *device_context, in FreeMemoryByRefCount()
400 … const DeviceContext *device_context, size_t alloc_size, in SetOpContextMemoryAllocFail()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_kernel_task.cc | 29 const device::DeviceContext *device_context) { in MallocMemoryForDeviceAddress()
42 … const device::DeviceContext *device_context, std::vector<int64_t> vec) { in MallocMemoryAndCopyValue()
63 auto device_context = context_->device_context(); in RunWithRet() local
116 auto device_context = context_->device_context(); in RunWithRet() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_kernel_task.cc | 24 const device::DeviceContext *device_context) { in MallocMemoryForDeviceAddress()
38 auto device_context = context_->device_context(); in RunWithRet() local
64 auto device_context = context_->device_context(); in RunWithRet() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/ |
| D | aclnn_utils.h | 28 #define DISPATCH_LAUNCH_KERNEL(device_context, aclnn_name, ws_ptr, ws_size, executor, stream, relea… argument
38 #define LAUNCH_ACLNN(aclnn_api, device_context, stream_id, ...) … argument
70 #define LAUNCH_ACLNN_SYNC(aclnn_api, device_context, stream_id, ...) … argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | silu.cc | 28 OpPtr SiLUCPUCall(const device::DeviceContext *device_context, const BaseTensorPtr &x_tensor) { in SiLUCPUCall()
40 auto device_context = op->device_context(); in SiLUCPUCustomize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | square.cc | 28 … SquareAscendCall(const std::shared_ptr<OpRunner> &op, const device::DeviceContext *device_context, in SquareAscendCall()
44 auto device_context = op->device_context(); in SquareAscendCustomize() local
|
| D | rand_like_ext.cc | 32 auto device_context = op->device_context(); in RandLikeExtAscendCustomize() local
40 auto device_context = op->device_context(); in RandLikeExtAscendCustomize() local
|
| D | prod_ext.cc | 49 auto device_context = op->device_context(); in ProdExtAscendCustomize() local
61 auto device_context = op->device_context(); in ProdExtAscendCustomize() local
|
| D | sigmoid_grad.cc | 27 …oidGradAscendCall(const std::shared_ptr<OpRunner> &op, const device::DeviceContext *device_context, in SigmoidGradAscendCall()
43 auto device_context = op->device_context(); in SigmoidGradAscendCustomize() local
|
| D | rand_ext.cc | 36 auto device_context = op->device_context(); in RandExtAscendCustomize() local
42 auto device_context = op->device_context(); in RandExtAscendCustomize() local
|
| D | tanh_grad.cc | 27 …anhGradAscendCall(const std::shared_ptr<OpRunner> &op, const device::DeviceContext *device_context, in TanhGradAscendCall()
43 auto device_context = op->device_context(); in TanhGradAscendCustomize() local
|
| D | masked_fill.cc | 29 const device::DeviceContext *device_context, in MaskedFillAscendCall()
46 auto device_context = op->device_context(); in MaskedFillAscendCustomize() local
|
| D | grid_sampler_2d_grad.cc | 36 auto device_context = op->device_context(); in GridSampler2DGradAscendCustomize() local
56 auto device_context = op->device_context(); in GridSampler2DGradAscendCustomize() local
|
| D | softmax.cc | 28 …SoftmaxAscendCall(const std::shared_ptr<OpRunner> &op, const device::DeviceContext *device_context, in SoftmaxAscendCall()
50 auto device_context = op->device_context(); in SoftmaxAscendCustomize() local
|
| D | grid_sampler_3d_grad.cc | 36 auto device_context = op->device_context(); in GridSampler3DGradAscendCustomize() local
56 auto device_context = op->device_context(); in GridSampler3DGradAscendCustomize() local
|
| D | relu_grad.cc | 27 …eluGradAscendCall(const std::shared_ptr<OpRunner> &op, const device::DeviceContext *device_context, in ReluGradAscendCall()
43 auto device_context = op->device_context(); in ReluGradAscendCustomize() local
|
| D | upsample_nearest1d.cc | 27 const device::DeviceContext *device_context, in UpsampleNearest1dAscendCall()
52 auto device_context = op->device_context(); in UpsampleNearest1DAscendCustomize() local
|
| D | upsample_nearest2d.cc | 27 const device::DeviceContext *device_context, in UpsampleNearest2dAscendCall()
52 auto device_context = op->device_context(); in UpsampleNearest2DAscendCustomize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pipeline/task/ |
| D | kernel_task.h | 32 …KernelTaskContext(const device::DeviceContext *device_context, device::DeviceAddressPtrList input_… in KernelTaskContext()
40 const device::DeviceContext *device_context() { return device_context_; } in device_context() function
|