提交 cb59d1f6 authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/gpuarray/linalg.py

上级 ca9336ad
......@@ -7,7 +7,7 @@ import warnings
from theano import Op
from theano.gpuarray import basic_ops, GpuArrayType
import numpy
import numpy as np
from numpy.linalg.linalg import LinAlgError
try:
......@@ -107,7 +107,7 @@ class GpuCusolverSolve(Op):
ctx.cusolver_handle = cusolver.cusolverDnCreate()
def check_dev_info(self, dev_info):
val = numpy.asarray(dev_info)[0]
val = np.asarray(dev_info)[0]
if val > 0:
raise LinAlgError('A is singular')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论