提交 c1a888a6 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

add option to true using C code for GPU

上级 56a6097e
...@@ -4532,7 +4532,10 @@ class IncSubtensor(Op): ...@@ -4532,7 +4532,10 @@ class IncSubtensor(Op):
def c_code(self, node, name, inputs, outputs, sub): def c_code(self, node, name, inputs, outputs, sub):
if not isinstance(node.inputs[0].type, TensorType): if not isinstance(node.inputs[0].type, TensorType):
raise NotImplementedError() x = raw_input('Use GPU C code?')
if x == 'n':
raise NotImplementedError()
assert x == 'y'
if self.inplace: # convert bool to int if self.inplace: # convert bool to int
inplace = 1 inplace = 1
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论