Example #1
0
    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()
Example #2
0
	def setUpClass(self):
		self.env_orig = os.getenv('OPTKIT_USE_LOCALLIBS', '0')
		os.environ['OPTKIT_USE_LOCALLIBS'] = '1'
		self.libs = PogsLibs()
		self.A_test = self.A_test_gen