提交 494383ec authored 作者: Pascal Lamblin's avatar Pascal Lamblin 提交者: Frederic Bastien

Execute more things on GPU in tests

This should expose an issue in GpuIncSubtensor
上级 ecd18089
......@@ -545,8 +545,8 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin):
t = op(n[:z, :z], m)
gn, gm = theano.tensor.grad(theano.tensor.sum(t), [n, m])
utt.verify_grad(lambda m: op(n[:z, :z], m), [mv])
utt.verify_grad(lambda nn: op(nn[:z, :z], mv), [data])
utt.verify_grad(lambda m: op(n[:z, :z], m), [mv], mode=self.mode)
utt.verify_grad(lambda nn: op(nn[:z, :z], mv), [data], mode=self.mode)
def test_grad_0d(self):
data = np.asarray(rand(2, 3), dtype=self.dtype)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论