提交 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): ...@@ -75,10 +75,11 @@ class GpuElemwise(HideC, Elemwise):
pass pass
try: try:
support_code = self.scalar_op.c_support_code() support_code = self.scalar_op.c_support_code()
if (support_code.strip() != "#define THEANO_MACRO_MOD(x,y) (x % y)" and if "struct" in support_code:
support_code.strip() != ""):
# The macro is fine, the C++ struct is not. # 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: except MethodNotDefined:
pass pass
......
...@@ -946,7 +946,7 @@ def local_lift_abstractconv2d(node, context_name): ...@@ -946,7 +946,7 @@ def local_lift_abstractconv2d(node, context_name):
return [node.op(*inps)] return [node.op(*inps)]
# Register this here so that it goes after the abstract lifting # 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") @register_opt("low_memory")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论