Searched refs:GPUKernelRuntime (Results 1 – 5 of 5) sorted by relevance
65 bool GPUKernelRuntime::SyncStream() { in SyncStream()76 bool GPUKernelRuntime::Init() { in Init()216 bool GPUKernelRuntime::MemcpyAsync(void *dst, const void *src, uint64_t size, int32_t kind) { in MemcpyAsync()227 DeviceAddressPtr GPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress()232 DeviceAddressPtr GPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress()237 bool GPUKernelRuntime::InitDevice() { in InitDevice()262 void GPUKernelRuntime::ReleaseDeviceRes() { in ReleaseDeviceRes()298 void GPUKernelRuntime::ClearGraphRuntimeResource(uint32_t graph_id) { in ClearGraphRuntimeResource()303 void GPUKernelRuntime::AllocInplaceNodeMemory(const session::KernelGraph *graph) { in AllocInplaceNodeMemory()358 bool GPUKernelRuntime::IsDistributedTraining(const session::KernelGraph *graph) { in IsDistributedTraining()[all …]
39 class GPUKernelRuntime : public KernelRuntime {41 GPUKernelRuntime() = default;42 ~GPUKernelRuntime() override = default;65 GPUKernelRuntime(const GPUKernelRuntime &);66 GPUKernelRuntime &operator=(const GPUKernelRuntime &);140 MS_REG_KERNEL_RUNTIME(kGPUDevice, GPUKernelRuntime);
49 class GPUKernelRuntime; variable122 friend class mindspore::device::gpu::GPUKernelRuntime;
12 …rc/runtime/device/gpu/gpu_kernel_runtime.cc:mindspore::device::gpu::GPUKernelRuntime::LaunchKernel…
609 … auto gpu_runtime_instance = dynamic_cast<device::gpu::GPUKernelRuntime *>(runtime_instance); in UpdateOutputTensors()