Searched refs:kernel_addr (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | adam_delta_cpu_kernel_test.cc | 42 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
43 kernel_addr->addr = addr; in CreateKernelAddress()
44 kernel_addr->size = elem_num * 4; in CreateKernelAddress()
45 return kernel_addr; in CreateKernelAddress()
|
| D | unique_cpu_kernel_test.cc | 40 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
41 kernel_addr->addr = addr; in CreateKernelAddress()
42 return kernel_addr; in CreateKernelAddress()
|
| D | unique_with_pad_cpu_kernel_test.cc | 40 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
41 kernel_addr->addr = addr; in CreateKernelAddress()
42 return kernel_addr; in CreateKernelAddress()
|
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 46 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
47 kernel_addr->addr = addr; in CreateKernelAddress()
48 return kernel_addr; in CreateKernelAddress()
|
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 42 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
43 kernel_addr->addr = addr; in CreateKernelAddress()
44 return kernel_addr; in CreateKernelAddress()
|
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 42 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
43 kernel_addr->addr = addr; in CreateKernelAddress()
44 return kernel_addr; in CreateKernelAddress()
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 42 auto kernel_addr = std::make_shared<Address>(); in CreateKernelAddress() local
43 kernel_addr->addr = addr; in CreateKernelAddress()
44 return kernel_addr; in CreateKernelAddress()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_mod.cc | 113 CUfunction kernel_addr; in Launch() local
114 CUresult result = kernelmanager_->GetFunction(kernel_pack_, false, &thread_info, &kernel_addr); in Launch()
130 …result = cuLaunchKernel(kernel_addr, thread_info[0], thread_info[1], thread_info[2], thread_info[3… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.h | 153 … const DeviceAddress *device_address, const kernel::AddressPtr &kernel_addr);
|
| D | kernel_runtime.cc | 1287 … const DeviceAddress *device_address, const kernel::AddressPtr &kernel_addr) { in GetOrMallocAddress() argument
1289 kernel_addr->addr = device_address->ptr_; in GetOrMallocAddress()
1291 kernel_addr->addr = mem_scheduler->GetOrMalloc(device_address, device_address->size_); in GetOrMallocAddress()
1293 device_address->ptr_ = kernel_addr->addr; in GetOrMallocAddress()
|