提交 0ec4a230 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

remove perform method

上级 3d0b9980
...@@ -354,15 +354,6 @@ class GpuGemmBatch(BlasOp): ...@@ -354,15 +354,6 @@ class GpuGemmBatch(BlasOp):
assert A.dtype == B.dtype == C.dtype assert A.dtype == B.dtype == C.dtype
return Apply(self, [C, alpha, A, B, beta], [C.type()]) return Apply(self, [C, alpha, A, B, beta], [C.type()])
def perform(self, node, inputs, outputs):
C, alpha, A, B, beta = inputs
if self.inplace and C.flags.forc:
C *= beta
C += alpha * blas.batched_dot(A, B)
outputs[0][0] = C
else:
outputs[0][0] = alpha * blas.batched_dot(A, B) + beta * C
def c_headers(self): def c_headers(self):
return super(GpuGemmBatch, self).c_headers() + ['<gpuarray/blas.h>'] return super(GpuGemmBatch, self).c_headers() + ['<gpuarray/blas.h>']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论