def __init__(self, gpu=False, single_precision=False): self.version = None self.__device = None self.__precision = None self.__config = "(No libraries selected)" self.lowtypes = None self.dimcheck = getenv('OPTKIT_CHECKDIM', 0) self.typecheck = getenv('OPTKIT_CHECKTYPE', 0) self.devicecheck = getenv('OPTKIT_CHECKDEVICE', 0) # library loaders # self.dense_lib_loader = DenseLinsysLibs() # self.sparse_lib_loader = SparseLinsysLibs() # self.prox_lib_loader = ProxLibs() self.pogs_lib_loader = PogsLibs() self.cluster_lib_loader = ClusteringLibs() # library instances # self.dense = None # self.sparse = None # self.prox = None self.pogs = None self.cluster = None self.__LIBGUARD_ON = False self.__COBJECT_COUNT = 0 self.__set_lib()
def setUpClass(self): self.env_orig = os.getenv('OPTKIT_USE_LOCALLIBS', '0') os.environ['OPTKIT_USE_LOCALLIBS'] = '1' self.libs = ClusteringLibs()