提交 9e9dda28 authored 作者: Frederic Bastien's avatar Frederic Bastien

skip test if pycuda is not their.

上级 6fd7f9bf
import numpy import numpy
try:
import pycuda
except ImportError:
from nose.plugins.skip import SkipTest
raise SkipTest("Pycuda not installed. Skip test of theano op with pycuda code.")
import theano import theano
import theano.tensor as T import theano.tensor as T
from theano.misc.pycuda_example import PycudaElemwiseSourceModule, PycudaElemwiseKernel from theano.misc.pycuda_example import PycudaElemwiseSourceModule, PycudaElemwiseKernel
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论