Package org.apache.sysds.runtime.instructions.gpu.context
-
Interface Summary Interface Description GPUMemoryAllocator -
Class Summary Class Description CSRPointer Compressed Sparse Row (CSR) format for CUDA Generalized matrix multiply is implemented for CSR format in the cuSparse library among other operations Since we assume that the matrix is stored with zero-based indexing (i.e.CudaMemoryAllocator ExecutionConfig Java Wrapper to specify CUDA execution configuration for launching custom kernelsGPUContext Represents a context per GPU accessible through the same JVM.GPUContextPool GPULazyCudaFreeMemoryManager GPUMatrixMemoryManager GPUMemoryEviction GPUMemoryManager - All cudaFree and cudaMalloc in SystemDS should go through this class to avoid OOM or incorrect results.GPUMemoryManager.EvictionPolicyBasedComparator Class that governs the eviction policyGPUObject Handle to a matrix block on the GPUJCudaKernels ShadowBuffer UnifiedMemoryAllocator