提交 477b54a4 authored 作者: Caglar's avatar Caglar

changed the op's name.

上级 d467adf9
...@@ -7,7 +7,7 @@ from theano.tensor import as_tensor_variable ...@@ -7,7 +7,7 @@ from theano.tensor import as_tensor_variable
from scikits.cuda import cula from scikits.cuda import cula
def gpu_solve(A, b, trans='N'): def cula_gpu_solve(A, b, trans='N'):
cula.culaInitialize() cula.culaInitialize()
A_shape = A.shape A_shape = A.shape
b_shape = b.shape b_shape = b.shape
...@@ -96,7 +96,7 @@ class GpuSolve(GpuOp): ...@@ -96,7 +96,7 @@ class GpuSolve(GpuOp):
A_pycuda = to_gpuarray(A) A_pycuda = to_gpuarray(A)
b_pycuda = to_gpuarray(b) b_pycuda = to_gpuarray(b)
gpu_solve(A_pycuda, b_pycuda, self.trans) cula_gpu_solve(A_pycuda, b_pycuda, self.trans)
thunk.inputs = inputs thunk.inputs = inputs
thunk.outputs = outputs thunk.outputs = outputs
...@@ -104,4 +104,4 @@ class GpuSolve(GpuOp): ...@@ -104,4 +104,4 @@ class GpuSolve(GpuOp):
return thunk return thunk
#gpu_solve = GpuSolve() gpu_solve = GpuSolve()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论