GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 19:59:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Sharded the CUDA argmax tensor test

Browse Source
This commit is contained in:
Benoit Steiner 2016-01-31 10:44:15 -08:00
parent 483082ef6e
commit 4a2ddfb81d
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
unsupported/test/cxx11_tensor_argmax_cuda.cu
View File

@ -242,10 +242,10 @@ void test_cuda_argmin_dim()
void test_cxx11_tensor_cuda() void test_cxx11_tensor_cuda()
{ {
CALL_SUBTEST(test_cuda_simple_argmax<RowMajor>()); CALL_SUBTEST_1(test_cuda_simple_argmax<RowMajor>());
CALL_SUBTEST(test_cuda_simple_argmax<ColMajor>()); CALL_SUBTEST_1(test_cuda_simple_argmax<ColMajor>());
CALL_SUBTEST(test_cuda_argmax_dim<RowMajor>()); CALL_SUBTEST_2(test_cuda_argmax_dim<RowMajor>());
CALL_SUBTEST(test_cuda_argmax_dim<ColMajor>()); CALL_SUBTEST_2(test_cuda_argmax_dim<ColMajor>());
CALL_SUBTEST(test_cuda_argmin_dim<RowMajor>()); CALL_SUBTEST_3(test_cuda_argmin_dim<RowMajor>());
CALL_SUBTEST(test_cuda_argmin_dim<ColMajor>()); CALL_SUBTEST_3(test_cuda_argmin_dim<ColMajor>());
} }