def testCuVectorSwap(self):
N = [2, 3, 5, 7, 13]
A = Vector(N).clone()
C = CuVector.from_size(5)
C.swap(A) #Swap *is* destructive
self.assertEqual(16.0, C.norm(2))
A = Vector()
C = CuVector.from_size(0)
C.swap(A)
self.assertEqual(0.0, C.norm(2))
def testCuVectorInvertElements(self):
# Test that this doesnt crash. This crashes when CUDA is enabled.
# C = CuVector.from_size(0)
# C.invert_elements()
C = CuVector.from_size(10)
C.set_randn()
C.invert_elements()
# Geometric series r = 1/2, a = 1/2
A = Vector([2, 4, 8, 16, 32, 64])
C = CuVector.from_size(len(A))
C.swap(A)
C.invert_elements()
f1 = C.sum()
self.assertAlmostEqual(0.5 * (1 - 0.5**len(A)) / (1 - 0.5), f1)
def testCuVectorNewFromSize(self):
vec = CuVector()
self.assertIsNotNone(vec)
self.assertEqual(0, vec.dim())
for i in range(10):
dim = 10 * i
vec = CuVector.from_size(dim)
self.assertIsNotNone(vec)
self.assertEqual(dim, vec.dim())
def testCuVectorCopyFromVec(self):
# Shouldnt crash
A = Vector()
C = CuVector.from_size(0)
C.copy_from_vec(A)
# What if dims not match?
# HARD-CRASH
# FIXME
# A = Vector.random(10)
# C = CuVector.from_size(0)
# C.CopyFromVec(A)
for i in range(10):
dim = 10 * i
A = Vector(dim)
A.set_randn_()
D = CuVector.from_size(dim)
D.copy_from_vec(A)
self.assertEqual(A.sum(), D.sum())
def testCuSubVector(self):
for iteration in range(10):
M1 = 1 + kaldi_math.rand() % 10
M2 = 1 + kaldi_math.rand() % 1
M3 = 1 + kaldi_math.rand() % 10
M = M1 + M2 + M3
m = kaldi_math.rand() % M2
vec = CuVector.from_size(M)
vec.set_randn()
subvec1 = CuSubVector(vec, M1, M2)
# subvec2 = vec.range(M1, M2)
f1, f2, f3 = vec[M1 + m], subvec1[m], subvec2[m]
self.assertEqual(f1, f2)
self.assertEqual(f3, f2)
