提交 3f3cdb9a authored 作者: notoraptor's avatar notoraptor

Update GpuDimShuffle.perform(). Class itself will be parameterized in another PR.

上级 5ad8bbc2
...@@ -430,7 +430,7 @@ class GpuDimShuffle(HideC, DimShuffle): ...@@ -430,7 +430,7 @@ class GpuDimShuffle(HideC, DimShuffle):
s = "GpuDimShuffle{%s}" s = "GpuDimShuffle{%s}"
return s % (','.join(str(x) for x in self.new_order)) return s % (','.join(str(x) for x in self.new_order))
def perform(self, node, inp, out): def perform(self, node, inp, out, params):
input, = inp input, = inp
storage, = out storage, = out
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论