提交 95aa14cd authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add the missing puzzle piece.

上级 0a5d132c
...@@ -188,15 +188,17 @@ def check_init_y(): ...@@ -188,15 +188,17 @@ def check_init_y():
if not have_fblas: if not have_fblas:
check_init_y._result = False check_init_y._result = False
gemv = _blas_gemv_fns['float64']
y = float('NaN') * numpy.ones((2,)) y = float('NaN') * numpy.ones((2,))
x = numpy.ones((2,)) x = numpy.ones((2,))
A = numpy.ones((2, 2)) A = numpy.ones((2, 2))
gemv = _blas_gemv_fns[y.dtype]
gemv(1.0, A.T, x, 0.0, y, overwrite_y=True, trans=True) gemv(1.0, A.T, x, 0.0, y, overwrite_y=True, trans=True)
check_init_y._result = numpy.isnan(y).any() check_init_y._result = numpy.isnan(y).any()
return check_init_y._result return check_init_y._result
check_init_y._result = None
class Gemv(Op): class Gemv(Op):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论