提交 38e2e460 authored 作者: Ramana Subramanyam's avatar Ramana Subramanyam 提交者: GitHub

Merge pull request #3 from nouiz/Sentient07-CrashFix#1

A few other fix to make the SBRNN code compile in the new back-end in fast_compile
......@@ -75,10 +75,11 @@ class GpuElemwise(HideC, Elemwise):
pass
try:
support_code = self.scalar_op.c_support_code()
if (support_code.strip() != "#define THEANO_MACRO_MOD(x,y) (x % y)" and
support_code.strip() != ""):
if "struct" in support_code:
# The macro is fine, the C++ struct is not.
raise SupportCodeError(support_code)
raise SupportCodeError(
"struct aren't supported in GpuElemwise support_code" +
support_code)
except MethodNotDefined:
pass
......
......@@ -946,7 +946,7 @@ def local_lift_abstractconv2d(node, context_name):
return [node.op(*inps)]
# Register this here so that it goes after the abstract lifting
register_opt()(conv_groupopt)
register_opt('fast_compile')(conv_groupopt)
@register_opt("low_memory")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论