提交 831bfcd3 authored 作者: nouiz's avatar nouiz

Merge pull request #183 from jaberg/smallfix

Smallfix
...@@ -43,7 +43,7 @@ class NaiveAlgo(object): ...@@ -43,7 +43,7 @@ class NaiveAlgo(object):
:param sync: if True, will wait after the kernel launch and check for error call. :param sync: if True, will wait after the kernel launch and check for error call.
""" """
try: try:
code = scalar_op.c_support_code_apply(node=None, name="nodename") code = scalar_op.c_support_code_apply(None, "nodename")
if code: if code:
raise SupportCodeError(scalar_op) raise SupportCodeError(scalar_op)
except gof.utils.MethodNotDefined: except gof.utils.MethodNotDefined:
......
...@@ -1171,7 +1171,8 @@ class TestCompositeCodegen(unittest.TestCase): ...@@ -1171,7 +1171,8 @@ class TestCompositeCodegen(unittest.TestCase):
y = self.times_2(self.x) y = self.times_2(self.x)
z = self.times_3(y) z = self.times_3(y)
f = theano.function([self.x], cuda.gpu_from_host(z)) f = theano.function([self.x], cuda.gpu_from_host(z),
mode=theano.compile.mode.get_default_mode().including('gpu'))
topo = f.maker.env.toposort() topo = f.maker.env.toposort()
assert len(topo) == 2 assert len(topo) == 2
assert topo[1].op == cuda.gpu_from_host assert topo[1].op == cuda.gpu_from_host
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论