org.apache.sysml.runtime.instructions.gpu.context

Class GPUContext

    • Field Detail

      • currContext

        protected static GPUContext currContext
      • isGPUContextCreated

        public static volatile Boolean isGPUContextCreated
    • Constructor Detail

      • GPUContext

        protected GPUContext()
    • Method Detail

      • getAvailableMemory

        public abstract long getAvailableMemory()
        Gets device memory available for SystemML operations
        Returns:
        available memory
      • ensureComputeCapability

        public abstract void ensureComputeCapability()
                                              throws DMLRuntimeException
        Ensures that all the CUDA cards on the current system are of the minimum required compute capability. (The minimum required compute capability is hard coded in JCudaContext.
        Throws:
        DMLRuntimeException - if DMLRuntimeException occurs
      • createGPUObject

        public static GPUObject createGPUObject(org.apache.sysml.runtime.controlprogram.caching.MatrixObject mo)

Copyright © 2017 The Apache Software Foundation. All rights reserved.