Package | Description |
---|---|
org.apache.sysml.runtime.instructions.gpu.context | |
org.apache.sysml.runtime.matrix.data |
Class and 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.
|
ExecutionConfig
Java Wrapper to specify CUDA execution configuration for launching custom kernels
|
GPUContext
Represents a context per GPU accessible through the same JVM.
|
GPULazyCudaFreeMemoryManager |
GPUMatrixMemoryManager |
GPUMemoryAllocator |
GPUMemoryManager
- All cudaFree and cudaMalloc in SystemML should go through this class to avoid OOM or incorrect results.
|
GPUObject
Handle to a matrix block on the GPU
|
JCudaKernels
Utility class that allows LibMatrixCUDA as well as GPUObject to invoke custom CUDA kernels.
|
Class and 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.
|
GPUContext
Represents a context per GPU accessible through the same JVM.
|
JCudaKernels
Utility class that allows LibMatrixCUDA as well as GPUObject to invoke custom CUDA kernels.
|
Copyright © 2018 The Apache Software Foundation. All rights reserved.