Searched refs:GpuKernel (Results 1 – 7 of 7) sorted by relevance
36 class GpuKernel : public internal::KernelInterface {38 GpuKernel() in GpuKernel() function45 ~GpuKernel() override {} in ~GpuKernel()92 inline const GpuKernel* AsGpuKernel(const KernelBase* kernel) { in AsGpuKernel()93 return static_cast<const GpuKernel*>(kernel->implementation()); in AsGpuKernel()98 inline GpuKernel* AsGpuKernel(KernelBase* kernel) { in AsGpuKernel()99 return static_cast<GpuKernel*>(kernel->implementation()); in AsGpuKernel()
264 bool GetKernelMetadata(GpuKernel* cuda_kernel,
30 using CUDAKernel = gpu::GpuKernel;
21 CUfunc_cache GpuKernel::GetGpuCacheConfig() const { in GetGpuCacheConfig()
269 GpuKernel* cuda_kernel = AsGpuKernel(kernel); in GetKernel()402 bool GpuExecutor::GetKernelMetadata(GpuKernel* cuda_kernel, in GetKernelMetadata()425 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in Launch()487 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in VlogOccupancyInfo()946 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()
21 hipFuncCache_t GpuKernel::GetGpuCacheConfig() const { in GetGpuCacheConfig()
223 GpuKernel* rocm_kernel = AsGpuKernel(kernel); in GetKernel()274 bool GpuExecutor::GetKernelMetadata(GpuKernel* rocm_kernel, in GetKernelMetadata()291 const GpuKernel* rocm_kernel = AsGpuKernel(&kernel); in Launch()848 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()