提交 9766f81a authored 作者: Frederic's avatar Frederic

fix test in FAST_COMPILE

上级 5ec4c302
...@@ -199,6 +199,7 @@ def test_dnn_tag(): ...@@ -199,6 +199,7 @@ def test_dnn_tag():
class TestDnnInferShapes(utt.InferShapeTester): class TestDnnInferShapes(utt.InferShapeTester):
def setUp(self): def setUp(self):
super(TestDnnInferShapes, self).setUp() super(TestDnnInferShapes, self).setUp()
self.mode = mode_with_gpu
def test_softmax(self): def test_softmax(self):
t = T.ftensor4('t') t = T.ftensor4('t')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论