add sgvector to_gpu test

shogun-toolbox · Mar 13, 2017 · 26ea214 · 26ea214
1 parent 968745c
commit 26ea214
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/unit/mathematics/linalg/LinalgBackendViennaCL_unittest.cc b/tests/unit/mathematics/linalg/LinalgBackendViennaCL_unittest.cc
@@ -35,6 +35,22 @@ TEST(LinalgBackendViennaCL, SGVector_to_gpu_viennacl)
 	EXPECT_TRUE(b.on_gpu());
 }
 
+TEST(LinalgBackendViennaCL, SGVector_to_gpu_inplace_viennacl)
+{
+	sg_linalg->set_gpu_backend(new LinalgBackendViennaCL());
+
+	const index_t size = 10;
+	SGVector<int32_t> a(size), b(size);
+	a.range_fill(0);
+	to_gpu(a, a);
+	from_gpu(a, b);
+
+	EXPECT_TRUE(a.on_gpu());
+	ASSERT_FALSE(b.on_gpu());
+	for (index_t i = 0; i < size; ++i)
+		EXPECT_EQ(b[i], i);
+}
+
 TEST(LinalgBackendViennaCL, SGMatrix_to_gpu_viennacl)
 {
 	sg_linalg->set_gpu_backend(new LinalgBackendViennaCL());