• Home
• org.scalanlp
• gust_2.11
• 0.1
• source code
• CuDevice.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

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

package gust.util.cuda

import jcuda.driver.{JCudaDriver, CUdevice}

/**
 * TODO
 *
 * @author dlwh
 **/
class CuDevice(val device: CUdevice) {
  override def toString = device.toString
}

object CuDevice {
  implicit lazy val defaultDevice = {
    apply(0)
  }

  def apply(deviceNumber: Int):CuDevice = {
    JCudaDriver.cuInit(0)
    JCudaDriver.setExceptionsEnabled(true)
    val device = new CUdevice
    JCudaDriver.cuDeviceGet(device, deviceNumber)
    new CuDevice(device)
  }
}