Commit f9f5ee8
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -5,9 +5,6 @@ | |||
| 5 | 5 | import sys | |
| 6 | 6 | import unittest | |
| 7 | 7 | ||
| 8 | - import common | ||
| 9 | - import pycuda.driver | ||
| 10 | - | ||
| 11 | 8 | from common import gpu_test | |
| 12 | 9 | ||
| 13 | 10 | ||
@@ -19,7 +16,8 @@ def test_system_management_interface(self): | |||
| 19 | 16 | self.assertEqual(0, smi.returncode) | |
| 20 | 17 | ||
| 21 | 18 | @gpu_test | |
| 22 | - def test_pycuda(self): | ||
| 19 | + def test_pycuda(self): | ||
| 20 | + import pycuda.driver | ||
| 23 | 21 | pycuda.driver.init() | |
| 24 | 22 | gpu_name = pycuda.driver.Device(0).name() | |
| 25 | 23 | self.assertNotEqual(0, len(gpu_name)) | |
0 commit comments