org.apache.mxnet

Rtc

Related Doc: package mxnet

class Rtc extends AnyRef

This class allow you to write cuda kernel in Scala and call them with NDArray.

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. Rtc
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new Rtc(name: String, inputs: Array[(String, NDArray)], outputs: Array[(String, NDArray)], kernel: String)

    name

    String, name of the kernel.

    inputs

    Array[(String, NDArray)], array of input names and ndarray.

    outputs

    Array[(String, NDArray)], array of output names and ndarray.

    kernel

    String, the actual kernel code. Note that this is only the body of the kernel, i.e. after { and before }. Rtc will decorate the kernel. For example, if name = "mykernel" and inputs = Array(("x", NDArray.zeros(10))) outputs = Array(("y", NDArray.zeros(10))) kernel = "y[threadIdx.x] = x[threadIdx.x];", the kernel that is compile will be: extern "C" global mykernel(float *x, float *y) { const int x_ndim = 1; const int x_dims[] = { 10 }; const int y_ndim = 1; const int y_dims[] = { 10 };

    y[threadIdx.x] = x[threadIdx.x]; }

Value Members

  1. final def !=(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  5. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  6. def dispose(): Unit

    Free the rtc handle.

    Free the rtc handle. The object shall never be used after it is disposed.

  7. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  8. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  9. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  10. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  11. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  12. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  13. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  14. final def notify(): Unit

    Definition Classes
    AnyRef
  15. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  16. def push(ins: Array[NDArray], outs: Array[NDArray], gridDims: (Int, Int, Int), blockDims: (Int, Int, Int)): Unit

    run the kernel.

  17. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  18. def toString(): String

    Definition Classes
    AnyRef → Any
  19. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  20. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  21. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped