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)