提交 09122159 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix GpuAlloc using debug_perform from GpuAllocEmpty.

上级 44fbc1b4
......@@ -3768,6 +3768,10 @@ class GpuAlloc(GpuAllocEmpty):
shape, output = self.validate_shape(shape)
return Apply(self, [v] + shape, [output])
# This is required because the superclass (GpuAllocEmpty) also has it.
def debug_perform(self, node, inputs, out_):
self.perform(node, inputs, out_)
def perform(self, node, inputs, out_):
# the super class (GpuAllocEmpty) allocates memory, we fill it
value = inputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论