提交 1d2a81bb authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix wrong indices order in GpuSubtensor c_code().

上级 77a62e79
...@@ -108,7 +108,7 @@ class GpuSubtensor(HideC, Subtensor): ...@@ -108,7 +108,7 @@ class GpuSubtensor(HideC, Subtensor):
elif isinstance(idx, (numpy.integer, int)): elif isinstance(idx, (numpy.integer, int)):
return str(idx), 0 return str(idx), 0
elif isinstance(idx, gof.Type): elif isinstance(idx, gof.Type):
return indices.pop(), 0 return indices.pop(0), 0
else: else:
assert 0, idx assert 0, idx
...@@ -131,7 +131,7 @@ class GpuSubtensor(HideC, Subtensor): ...@@ -131,7 +131,7 @@ class GpuSubtensor(HideC, Subtensor):
fail=sub['fail'], inp=inp) fail=sub['fail'], inp=inp)
else: else:
if isinstance(idx, gof.Type): if isinstance(idx, gof.Type):
start = indices.pop() start = indices.pop(0)
elif isinstance(idx, (numpy.integer, int)): elif isinstance(idx, (numpy.integer, int)):
start = idx start = idx
else: else:
...@@ -153,4 +153,4 @@ class GpuSubtensor(HideC, Subtensor): ...@@ -153,4 +153,4 @@ class GpuSubtensor(HideC, Subtensor):
return sio.getvalue() return sio.getvalue()
def c_code_cache_version(self): def c_code_cache_version(self):
return (4,) return (5,)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论