提交 0e4f1777 authored 作者: Frederic's avatar Frederic

Enable GpuElemwise test for c code.

上级 742a6c8e
...@@ -15,11 +15,14 @@ from theano.sandbox.gpuarray.type import GpuArrayType ...@@ -15,11 +15,14 @@ from theano.sandbox.gpuarray.type import GpuArrayType
from pygpu.array import gpuarray from pygpu.array import gpuarray
# This is acutally a test for GpuElemwise # This is acutally a test for GpuElemwise
class test_gpu_Broadcast(test_Broadcast): class test_gpu_Broadcast(test_Broadcast):
op = GpuElemwise op = GpuElemwise
type = GpuArrayType type = GpuArrayType
cop = GpuElemwise
ctype = GpuArrayType
def rand_val(self, shp): def rand_val(self, shp):
return rand_gpuarray(*shp, **dict(cls=gpuarray)) return rand_gpuarray(*shp, **dict(cls=gpuarray))
...@@ -27,8 +30,8 @@ class test_gpu_Broadcast(test_Broadcast): ...@@ -27,8 +30,8 @@ class test_gpu_Broadcast(test_Broadcast):
#cop = GpuElemwise #cop = GpuElemwise
#ctype = GpuArrayType #ctype = GpuArrayType
#def rand_cval(self, shp): def rand_cval(self, shp):
# return rand_gpuarray(*shp, **dict(cls=gpuarray)) return rand_gpuarray(*shp, **dict(cls=gpuarray))
class test_GpuDimShuffle(test_DimShuffle): class test_GpuDimShuffle(test_DimShuffle):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论