def run_tests(self):
from kaldi.cudamatrix import cuda_available
if cuda_available():
from kaldi.cudamatrix import CuDevice
CuDevice.instantiate().set_debug_stride_mode(True)
CuDevice.instantiate().select_gpu_id("yes")
super(test_cuda, self).run_tests()
CuDevice.instantiate().print_profile()
else:
print("CUDA not available. Running tests on CPU.")
super(test_cuda, self).run_tests()
@unittest.skip("Not a sequential object")
def testCuVectorGetItem(self):
v = CuVector()
with self.assertRaises(IndexError):
v[0]
v = CuVector.new([3, 5, 7, 11, 13])
self.assertAlmostEqual(3.0, v[0])
self.assertAlmostEqual(7.0, v[2])
self.assertAlmostEqual(13.0, v[4])
with self.assertRaises(IndexError):
v[5]
if __name__ == '__main__':
if cuda_available():
for i in range(2):
CuDevice.Instantiate().SetDebugStrideMode(True)
if i == 0:
CuDevice.Instantiate().SelectGpuId("no")
else:
CuDevice.Instantiate().SelectGpuId("yes")
unittest.main()
CuDevice.Instantiate().PrintProfile()
else:
unittest.main()