Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_multivariate_normal_batch_correlated_samples_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_multivariate_normal_batch_correlated_samples(cuda=True)
def test_gauss_hermite_quadrature_1D_normal_nonbatch_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_gauss_hermite_quadrature_1D_normal_nonbatch(cuda=True)
def test_gauss_hermite_quadrature_1D_mvn_batch_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_gauss_hermite_quadrature_1D_normal_nonbatch(cuda=True)
def test_gauss_hermite_quadrature_1D_mvn_nonbatch_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_gauss_hermite_quadrature_1D_normal_nonbatch(cuda=True)
def test_multitask_multivariate_normal_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_multitask_multivariate_normal(cuda=True)
def test_normal_prior_batch_log_prob_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
return self.test_normal_prior_batch_log_prob(cuda=True)
def test_from_independent_mvns_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_from_independent_mvns(cuda=True)
def test_multivariate_normal_prior_log_prob_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
return self.test_multivariate_normal_prior_log_prob(cuda=True)
def test_multivariate_normal_correlated_samples_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_multivariate_normal_correlated_samples(cuda=True)
def test_lkj_prior_log_prob_cuda(self):
if torch.cuda.is_available():
with least_used_cuda_device():
self.test_lkj_prior_log_prob(cuda=True)