提交 aeaa11a6 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

test for viewop on GPU

上级 0f5f6ea1
import numpy
import unittest
from nose.plugins.skip import SkipTest
import theano
mode_with_opt = theano.compile.mode.get_default_mode()
mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu')
def test_viewop_gpu():
from theano.sandbox import cuda
if cuda.cuda_available == False:
raise SkipTest('Optional package cuda disabled')
_x = theano.tensor.fvector('x')
x = cuda.gpu_from_host(_x)
out = theano.compile.function_module.ViewOp()(x)
f = theano.function([x],
[theano.compile.function_module.ViewOp()(x)],
mode=mode_with_gpu)
data = numpy.array([1, 2, 3], dtype='float32')
assert numpy.allclose(f(data), data)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论