提交 e4c39c23 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Execute more things on GPU in tests

上级 e648ddeb
...@@ -968,12 +968,14 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin): ...@@ -968,12 +968,14 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin):
utt.verify_grad( utt.verify_grad(
inc_slice(slice(1, 2, None), slice(None, None, None)), inc_slice(slice(1, 2, None), slice(None, None, None)),
(np.asarray([[0, 1], [2, 3], [4, 5.]]), (np.asarray([[0, 1], [2, 3], [4, 5.]]),
np.asarray([[9, 9.]]),)) np.asarray([[9, 9.]]),),
mode=self.mode)
# single element # single element
utt.verify_grad( utt.verify_grad(
inc_slice(2, 1), inc_slice(2, 1),
(np.asarray([[0, 1], [2, 3], [4, 5.]]), np.asarray(9.),)) (np.asarray([[0, 1], [2, 3], [4, 5.]]), np.asarray(9.),),
mode=self.mode)
def test_inc_and_set_subtensor(self): def test_inc_and_set_subtensor(self):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论