Home
last modified time | relevance | path

Searched refs:CUDAKernel (Results 1 – 3 of 3) sorted by relevance

/external/llvm-project/polly/tools/GPURuntime/
DGPUJIT.c1252 if (((CUDAKernel *)Kernel->Kernel)->CudaModule) in freeKernelCUDA()
1253 CuModuleUnloadFcnPtr(((CUDAKernel *)Kernel->Kernel)->CudaModule); in freeKernelCUDA()
1256 free((CUDAKernel *)Kernel->Kernel); in freeKernelCUDA()
1274 ((CUDAKernel *)KernelCache[i]->Kernel)->BinaryString == BinaryBuffer) { in getKernelCUDA()
1285 Function->Kernel = (CUDAKernel *)malloc(sizeof(CUDAKernel)); in getKernelCUDA()
1341 Res = CuModuleLoadDataFcnPtr(&(((CUDAKernel *)Function->Kernel)->CudaModule), in getKernelCUDA()
1348 Res = CuModuleGetFunctionFcnPtr(&(((CUDAKernel *)Function->Kernel)->Cuda), in getKernelCUDA()
1349 ((CUDAKernel *)Function->Kernel)->CudaModule, in getKernelCUDA()
1358 ((CUDAKernel *)Function->Kernel)->BinaryString = BinaryBuffer; in getKernelCUDA()
1412 CuLaunchKernelFcnPtr(((CUDAKernel *)Kernel->Kernel)->Cuda, GridDimX, in launchKernelCUDA()
DGPUJIT.h95 typedef struct CUDAKernelT CUDAKernel; typedef
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_kernel.h30 using CUDAKernel = gpu::GpuKernel; variable