def setUp(self):
if not cusolver.check_availability('gesv'):
pytest.skip('gesv is not available')
self.dtype = numpy.dtype(self.dtype)
self.r_dtype = self.dtype.char.lower()
a = self._make_well_conditioned_matrix((self.n, self.n))
if self.nrhs is None:
x_shape = (self.n, )
else:
x_shape = (self.n, self.nrhs)
self.x_ref = self._make_random_matrix(x_shape, cupy)
b = numpy.dot(a, self.x_ref)
self.tol = self._tol[self.r_dtype]
self.a = cupy.array(a)
self.b = cupy.array(b)
if self.compute_type is not None:
self.old_compute_type = _linalg.get_compute_type(self.dtype)
_linalg.set_compute_type(self.dtype, self.compute_type)
def tearDown(self):
if self.compute_type is not None:
_linalg.set_compute_type(self.dtype, self.old_compute_type)
def setUp(self):
if not cusolver.check_availability('gels'):
pytest.skip('gels is not available')
if self.compute_type is not None:
self.old_compute_type = _linalg.get_compute_type(self.dtype)
_linalg.set_compute_type(self.dtype, self.compute_type)
