PLOOP.py

 
 
import numpy as np
from . import config_setter_utils as csu
 
 
 
class Param_loop:
 
    def __init__(self):
        self.__niter__niter = 0
        self.__ittime__ittime = 0.
        self.__devices__devices = np.array([0], dtype=np.int32)
 
    def get_devices(self):
        """ Get the list of GPU devices used
 
        :return: (np.ndarray[ndim=1, dtype=np.int32_t]) : list of GPU devices
        """
        return self.__devices__devices
 
    def set_devices(self, devices):
        """ Set the list of GPU devices used
 
        Args:
            devices: (np.ndarray[ndim=1, dtype=np.int32_t]) : list of GPU devices
        """
        self.__devices__devices = csu.enforce_array(devices, len(devices), dtype=np.int32,
                                           scalar_expand=False)
 
    devices = property(get_devices, set_devices)
 
    def get_niter(self):
        """ Get the number of iteration
 
        :return: (long) : number of iteration
        """
        return self.__niter__niter
 
    def set_niter(self, n):
        """ Set the number of iteration
 
        Args:
            n: (long) : number of iteration
        """
        self.__niter__niter = csu.enforce_int(n)
 
    niter = property(get_niter, set_niter)
 
    def get_ittime(self):
        """ Get iteration time
 
        :return: (float) :iteration time
        """
        return self.__ittime__ittime
 
    def set_ittime(self, t):
        """ Set iteration time
 
        Args:
            t: (float) :iteration time
        """
        self.__ittime__ittime = csu.enforce_float(t)
 
    ittime = property(get_ittime, set_ittime)