saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ffi tests skip on cuda

Browse source
This commit is contained in:
soumith 2016-10-03 12:15:28 -07:00
parent 64dd1419c5
commit 33371c5164
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
test/test_utils.py
View file

@ -5,16 +5,18 @@ import shutil
import random
import tempfile
import unittest
import torch
import torch.cuda
import sys
import traceback
import torch
import torch.cuda
from torch.autograd import Variable
from torch.utils.trainer import Trainer
from torch.utils.trainer.plugins import *
from torch.utils.trainer.plugins.plugin import Plugin
from torch.utils.data import *
HAS_CUDA = torch.cuda.is_available()
from common import TestCase
try:
@ -326,7 +328,7 @@ class TestFFI(TestCase):
self.assertRaises(torch.FatalError,
lambda: cpulib.bad_func(tensor, 2, 1.5))
@unittest.skipIf(not HAS_CFFI, "ffi tests require cffi package")
@unittest.skipIf(not HAS_CFFI or not HAS_CUDA, "ffi tests require cffi package")
def test_gpu(self):
compile_extension(
name='gpulib',