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()