[TEST] fix new test with cuda.