Skip to content

Commit 96b6c6e

Browse files
kyleabeauchamptwiecki
authored andcommitted
Skip logdet test on GPU
1 parent 6b39941 commit 96b6c6e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pymc3/tests/test_math.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from theano.tests import unittest_tools as utt
44
from pymc3.math import LogDet, logdet, probit, invprobit
55
from .helpers import SeededTest
6+
import pytest
67

78

89
def test_probit():
@@ -30,6 +31,7 @@ def validate(self, input_mat):
3031
# Test gradient:
3132
utt.verify_grad(self.op, [input_mat])
3233

34+
@pytest.mark.skipif(theano.config.device in ["cuda", "gpu"], reason="No logDet implementation on GPU.")
3335
def test_basic(self):
3436
# Calls validate with different params
3537
test_case_1 = np.random.randn(3, 3) / np.sqrt(3)

0 commit comments

Comments
 (0)