提交 4c6e3fc5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix crash in FAST_COMPILE.

上级 2ffd049c
...@@ -245,7 +245,7 @@ class Gemv(Op): ...@@ -245,7 +245,7 @@ class Gemv(Op):
raise TypeError('gemv requires vector for y', y.type) raise TypeError('gemv requires vector for y', y.type)
return Apply(self, [y, alpha, A, x, beta], [y.type()]) return Apply(self, [y, alpha, A, x, beta], [y.type()])
def perform(self, node, inputs, out_storage): def perform(self, node, inputs, out_storage, params=None):
y, alpha, A, x, beta = inputs y, alpha, A, x, beta = inputs
if (have_fblas and y.shape[0] != 0 and x.shape[0] != 0 and if (have_fblas and y.shape[0] != 0 and x.shape[0] != 0 and
y.dtype in _blas_gemv_fns): y.dtype in _blas_gemv_fns):
...@@ -335,7 +335,7 @@ class Ger(Op): ...@@ -335,7 +335,7 @@ class Ger(Op):
raise TypeError('only float and complex types supported', x.dtype) raise TypeError('only float and complex types supported', x.dtype)
return Apply(self, [A, alpha, x, y], [A.type()]) return Apply(self, [A, alpha, x, y], [A.type()])
def perform(self, node, inp, out): def perform(self, node, inp, out, params=None):
cA, calpha, cx, cy = inp cA, calpha, cx, cy = inp
cZ, = out cZ, = out
if self.destructive: if self.destructive:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论