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

file theano/gpuarray/linalg.py

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