COMPASS  5.4.4
End-to-end AO simulation tool using GPU acceleration
carmaWrap.context Class Reference
Collaboration diagram for carmaWrap.context:

Public Member Functions

None activate_tensor_cores (self, bool flag)
 Set the cublas math mode using tensor cores or not. More...
 
device get_device (self, int arg0)
 
int set_active_device (self, int arg0)
 
int set_active_device_force (self, int arg0)
 

Static Public Member Functions

context get_instance ()
 
context get_instance_1gpu (int arg0)
 
context get_instance_ngpu (int arg0, numpy.ndarray[numpy.int32] arg1)
 

Properties

 activeRealDevice = property
 
type of : int More...
 
 active_device = property
 
type of : int More...
 
 cudaRuntimeGetVersion = property
 
type of : int More...
 
 driverVersion = property
 
type of : int More...
 
 ndevice = property
 
type of : int More...
 
 ndevices = property
 
type of : int More...
 

Detailed Description

Definition at line 170 of file carmaWrap.py.

Member Function Documentation

◆ activate_tensor_cores()

None carmaWrap.context.activate_tensor_cores (   self,
bool  flag 
)

Set the cublas math mode using tensor cores or not.

Definition at line 174 of file carmaWrap.py.

◆ get_device()

device carmaWrap.context.get_device (   self,
int  arg0 
)

Definition at line 175 of file carmaWrap.py.

◆ get_instance()

context carmaWrap.context.get_instance ( )
static

Definition at line 177 of file carmaWrap.py.

◆ get_instance_1gpu()

context carmaWrap.context.get_instance_1gpu ( int  arg0)
static

Definition at line 179 of file carmaWrap.py.

◆ get_instance_ngpu()

context carmaWrap.context.get_instance_ngpu ( int  arg0,
numpy.ndarray[numpy.int32]  arg1 
)
static

Definition at line 181 of file carmaWrap.py.

◆ set_active_device()

int carmaWrap.context.set_active_device (   self,
int  arg0 
)

Definition at line 182 of file carmaWrap.py.

◆ set_active_device_force()

int carmaWrap.context.set_active_device_force (   self,
int  arg0 
)

Definition at line 183 of file carmaWrap.py.

Property Documentation

◆ active_device

carmaWrap.context.active_device = property
static


type of : int

Definition at line 195 of file carmaWrap.py.

◆ activeRealDevice

carmaWrap.context.activeRealDevice = property
static


type of : int

Definition at line 188 of file carmaWrap.py.

◆ cudaRuntimeGetVersion

carmaWrap.context.cudaRuntimeGetVersion = property
static


type of : int

Definition at line 202 of file carmaWrap.py.

◆ driverVersion

carmaWrap.context.driverVersion = property
static


type of : int

Definition at line 209 of file carmaWrap.py.

◆ ndevice

carmaWrap.context.ndevice = property
static


type of : int

Definition at line 216 of file carmaWrap.py.

◆ ndevices

carmaWrap.context.ndevices = property
static


type of : int

Definition at line 223 of file carmaWrap.py.


The documentation for this class was generated from the following file: