case class CudaDevice(i: Int) extends Device with Product with Serializable
- Alphabetic
- By Inheritance
- CudaDevice
- Serializable
- Product
- Equals
- Device
- AnyRef
- Any
- Hide All
- Show All
- Public
- Protected
Instance Constructors
- new CudaDevice(i: Int)
Value Members
- final def !=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- final def ##: Int
- Definition Classes
- AnyRef → Any
- final def ==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- def allocateBuffers[S](size: Long, options: STenOptions)(implicit arg0: Sc[S]): BufferPair
- Definition Classes
- Device
- final def asInstanceOf[T0]: T0
- Definition Classes
- Any
- def clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.CloneNotSupportedException]) @IntrinsicCandidate() @native()
- final def eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- final def getClass(): Class[_ <: AnyRef]
- Definition Classes
- AnyRef → Any
- Annotations
- @IntrinsicCandidate() @native()
- def getCurrentStream: CudaStream
Effective for the current OS thread
- def getStreamFromPool: CudaStream
- val i: Int
- final def isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- def measureTime[A](f: => A): (A, Long)
- Definition Classes
- CudaDevice → Device
- final def ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- final def notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @IntrinsicCandidate() @native()
- final def notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @IntrinsicCandidate() @native()
- def options[S](precision: FloatingPointPrecision)(implicit arg0: Sc[S]): STenOptions
- Definition Classes
- CudaDevice → Device
- def productElementNames: Iterator[String]
- Definition Classes
- Product
- def setCurrentStream(s: CudaStream): Unit
Effective for the current OS thread
- def setSeed(seed: Long): Unit
- Definition Classes
- CudaDevice → Device
- final def synchronized[T0](arg0: => T0): T0
- Definition Classes
- AnyRef
- def to(t: Tensor): Tensor
- Definition Classes
- CudaDevice → Device
- def to[S](t: STenOptions)(implicit arg0: Sc[S]): STenOptions
- Definition Classes
- CudaDevice → Device
- def to[S](t: STen)(implicit arg0: Sc[S]): STen
- Definition Classes
- Device
- def toBatched[S](tensors: Seq[STen], buffers: BufferPair)(implicit arg0: Sc[S]): Seq[STen]
Copies tensors to this device in a single cross device copy.
Copies tensors to this device in a single cross device copy. Data is copied via a buffer pair which consists of a source and a destinatin buffer. srcBuffer is supposed to be on the source device. dstBuffer has to be on
this
device. Tensors are first copied to the srcBuffer, then the srcBuffer is copied to dstBuffer, then the dstBuffer is split into views.All tensors must have the same data type.
Might make sense to pin the srcBuffer.
- Definition Classes
- Device
- final def wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException])
- final def wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException]) @native()
- final def wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException])
- def withOtherStream[A](synchronizeBefore: Boolean, synchronizeAfter: Boolean)(f: => A): A
Executes f on a new stream
Executes f on a new stream
f must not switch to other threads
Restores the stream to the original stream Optionally synchronizes the host before and/or after f
- Definition Classes
- CudaDevice → Device
Deprecated Value Members
- def finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.Throwable]) @Deprecated
- Deprecated
(Since version 9)