|
Nengo.ca | ||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |
java.lang.Objectca.nengo.util.impl.NEFGPUInterface
public class NEFGPUInterface
Allows running NEFEnsembles on the GPU. Passes the ensemble data to the GPU through a native function. Passes input to the GPU each step and stores the output from the GPU in the appropriate locations.
Field Summary | |
---|---|
protected float |
myEndTime
|
protected NEFEnsembleImpl[] |
myGPUEnsembles
|
protected Node[] |
myGPUNetworkArrays
|
protected Projection[] |
myGPUProjections
|
protected Node[] |
myNodes
|
protected Projection[] |
myProjections
|
protected static boolean |
myRequireAllOutputsOnCPU
|
protected float |
myStartTime
|
protected Projection[] |
nonGPUProjections
|
Constructor Summary | |
---|---|
NEFGPUInterface()
|
Method Summary | |
---|---|
static int[] |
findOptimalNodeAssignments(Node[] nodes,
Projection[] projections,
int numPartitions)
Used when there are multiple GPU's running a simulation. |
static java.lang.String |
getErrorMessage()
|
static int |
getNumAvailableDevices()
|
static int |
getRequestedNumDevices()
|
static boolean |
getUseGPU()
|
static void |
hideGPUTiming()
|
void |
initialize()
Gets all the necessary data from the nodes and projections which are assigned to run on GPUss and puts it in a form appropriate for passing to the native setup function. |
void |
kill()
|
static void |
requireAllOutputsOnCPU(boolean require)
|
static void |
setRequestedNumDevices(int value)
|
static void |
showGPUTiming()
|
void |
step(float startTime,
float endTime)
1. |
Node[] |
takeGPUNodes(Node[] nodes)
Finds all nodes in the given array which are supposed to execute on the GPU. |
Projection[] |
takeGPUProjections(Projection[] projections)
Finds all projections in the given array which are supposed to execute on the GPU. |
Methods inherited from class java.lang.Object |
---|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
Field Detail |
---|
protected static boolean myRequireAllOutputsOnCPU
protected NEFEnsembleImpl[] myGPUEnsembles
protected Projection[] myGPUProjections
protected Projection[] nonGPUProjections
protected Node[] myGPUNetworkArrays
protected Node[] myNodes
protected Projection[] myProjections
protected float myStartTime
protected float myEndTime
Constructor Detail |
---|
public NEFGPUInterface()
Method Detail |
---|
public static int getNumAvailableDevices()
public static void setRequestedNumDevices(int value)
public static int getRequestedNumDevices()
public static boolean getUseGPU()
public static void showGPUTiming()
public static void hideGPUTiming()
public static void requireAllOutputsOnCPU(boolean require)
public static java.lang.String getErrorMessage()
public void initialize()
public void step(float startTime, float endTime)
public void kill()
public static int[] findOptimalNodeAssignments(Node[] nodes, Projection[] projections, int numPartitions)
public Node[] takeGPUNodes(Node[] nodes)
public Projection[] takeGPUProjections(Projection[] projections)
|
Nengo.ca | ||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |