Skip to content

P
pytensor
提交 e1b2dc23 authored 作者: Shawn Tan's avatar Shawn Tan
浏览文件
操作

Removed node type check

上级 bcbbdf5e
显示空白字符变更
内嵌 并排
正在显示 包含 0 行增加 和 1 行删除
theano/gpuarray/opt.py
浏览文件 @ e1b2dc23
...@@ -1068,7 +1068,6 @@ def local_gpua_advanced_subtensor(op, context_name, inputs, outputs): ...@@ -1068,7 +1068,6 @@ def local_gpua_advanced_subtensor(op, context_name, inputs, outputs):
@op_lifter([tensor.AdvancedIncSubtensor1]) @op_lifter([tensor.AdvancedIncSubtensor1])
@register_opt2([tensor.AdvancedIncSubtensor1], 'fast_compile') @register_opt2([tensor.AdvancedIncSubtensor1], 'fast_compile')
def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs): def local_gpua_advanced_incsubtensor1(op, context_name, inputs, outputs):
if isinstance(op, (tensor.AdvancedIncSubtensor1)):
context = get_context(context_name) context = get_context(context_name)
# This is disabled on non-cuda contexts # This is disabled on non-cuda contexts
if context.kind != b'cuda': if context.kind != b'cuda':
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论