gust.util.cuda.CuContext.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gust_2.11 Show documentation

gust

The newest version!

package gust.util.cuda

import jcuda.driver.{CUdevice, CUcontext, JCudaDriver}
import jcuda.driver.JCudaDriver._

/**
 * TODO
 *
 * @author dlwh
 **/
class CuContext(val context: CUcontext) extends AnyVal {
  def withPush[T](t: =>T) = {
    cuCtxPushCurrent(context)
    try {
      t
    } finally {
      cuCtxPopCurrent(context)
    }

  }
}

object CuContext {
  def ensureContext(implicit device: CuDevice) = {

    // Try to obtain the current context
    var context: CUcontext = new CUcontext
    cuCtxGetCurrent(context)

    // If the context is 'null', then a new context
    // has to be created.
    val nullContext: CUcontext = new CUcontext

    if (context == nullContext) {
      context = new CUcontext
      cuCtxCreate(context, 0, device.device)
    }

    new CuContext(context)

  }
}