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