CudaStream

java.lang.Object

aten.CudaStream

public class CudaStream extends Object

Method Summary

Modifier and Type

Method

Description

static int

cudaGetDevice()

static void

cudaSetDevice(int device)

static CudaStream

getDefaultCUDAStream(byte device_index)

static CudaStream

getStreamFromPool(boolean isHighPriority, byte device_index)

static void

setCurrentCUDAStream(CudaStream stream)

void

synchronize()

String

toString()

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Method Details
- toString
  
  public String toString()
  
  Overrides:
  
  toString in class Object
- cudaSetDevice
  
  public static void cudaSetDevice(int device)
- cudaGetDevice
  
  public static int cudaGetDevice()
- synchronize
  
  public void synchronize()
- getStreamFromPool
  
  public static CudaStream getStreamFromPool(boolean isHighPriority, byte device_index)
- getDefaultCUDAStream
  
  public static CudaStream getDefaultCUDAStream(byte device_index)
- setCurrentCUDAStream
  
  public static void setCurrentCUDAStream(CudaStream stream)