org.nd4j.linalg.jcublas

Class SimpleJCublas

java.lang.Object

org.nd4j.linalg.jcublas.SimpleJCublas

public class SimpleJCublas
extends Object

Simple abstraction for jcublas operations

Author:

mjk, Adam Gibson

Constructor Summary

Constructors

Constructor and Description
SimpleJCublas()

Method Summary

Modifier and Type	Method and Description
static void	assertCudaBuffer(DataBuffer... buffer) Assert that the data buffer for each ndarray is a cuda buffer
static void	assertCudaBuffer(INDArray... buffer) Assert that the data buffer for each ndarray is a cuda buffer
static void	init() Initialize JCublas2.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SimpleJCublas

public SimpleJCublas()

Method Detail

assertCudaBuffer

public static void assertCudaBuffer(INDArray... buffer)
                             throws Exception

Assert that the data buffer for each ndarray is a cuda buffer

Parameters:

buffer - the arrays to tests

Throws:

Exception

assertCudaBuffer

public static void assertCudaBuffer(DataBuffer... buffer)
                             throws Exception

Assert that the data buffer for each ndarray is a cuda buffer

Parameters:

buffer - the arrays to tests

Throws:

Exception

init

public static void init()
                 throws Exception

Initialize JCublas2. Only called once

Throws:

Exception