naga.context.Context Class Reference

Python class for wrapping a CarmaContext. More...

Collaboration diagram for naga.context.Context:

Public Member Functions
def	__init__ (self, devices=None)
	Initializes a Context object. More...
def	getActiveDevice (self)
	Return the index of the current active device. More...
def	setActiveDevice (self, int index)
	Set the device index as the active device. More...
def	enableTensorCores (self)
	Enable the tensor cores math mode. More...
def	disableTensorCores (self)
	Disable the tensor cores math mode. More...

Public Attributes
	context
	devices

Detailed Description

Python class for wrapping a CarmaContext.

Definition at line 45 of file context.py.

Constructor & Destructor Documentation

__init__()

def naga.context.Context.__init__	(		self,
			devices = None
	)

Initializes a Context object.

   Parameters
   ------------

devices (list): (optionnal) list of devices to use. Default is 0

Definition at line 53 of file context.py.

        """
        if devices is None:
            self.context = context.get_instance_1gpu(0)
            self.devices = 0
        else:
            if isinstance(devices, list):
                devices = np.array(devices, dtype=np.int32)
            else:
                raise TypeError("Devices must be a list of integer")
            self.devices = devices
            self.context = context.get_instance_ngpu(devices)
 

Member Function Documentation

disableTensorCores()

def naga.context.Context.disableTensorCores

(

self

)

Disable the tensor cores math mode.

Definition at line 81 of file context.py.

        """ Disable the tensor cores math mode """
        self.context.activate_tensor_cores(True)

enableTensorCores()

def naga.context.Context.enableTensorCores

(

self

)

Enable the tensor cores math mode.

Definition at line 77 of file context.py.

        """ Enable the tensor cores math mode """
        self.context.activate_tensor_cores(True)
 

getActiveDevice()

def naga.context.Context.getActiveDevice

(

self

)

Return the index of the current active device.

Definition at line 66 of file context.py.

        """ Return the index of the current active device """
        return self.context.active_device
 

setActiveDevice()

def naga.context.Context.setActiveDevice	(		self,
		int	index
	)

Set the device index as the active device.

Definition at line 70 of file context.py.

        """ Set the device index as the active device """
        if index in self.devices:
            self.context.set_active_device(index)
        else:
            raise ValueError("Index given is not valid")
 

Member Data Documentation

context

naga.context.Context.context

Definition at line 55 of file context.py.

devices

naga.context.Context.devices

Definition at line 56 of file context.py.

---

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

• context.py