def lazyLib(): global lib if lib is None: from .util.gpu import device_count n_gpus, devices = device_count(n_gpus=-1) lib = lib_utils.get_lib(n_gpus, devices) return lib
def __init__(self, A, **kwargs): # Try to use all GPUs by default n_gpus = -1 for key, value in kwargs.items(): if key == "n_gpus": n_gpus = value # print("The value of {} is {}".format(key, value)) # sys.stdout.flush() from ..util.gpu import device_count n_gpus, devices = device_count(n_gpus=n_gpus) self.solver = BaseSolver(A, lib_utils.get_lib(n_gpus, devices)) assert self.solver is not None, "Couldn't instantiate Pogs Solver" self.info = self.solver.info self.solution = self.solver.pysolution