1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
/*
* GridTools
*
* Copyright (c) 2014-2023, ETH Zurich
* All rights reserved.
*
* Please, refer to the LICENSE file in the root directory.
* SPDX-License-Identifier: BSD-3-Clause
*/
#include <gridtools/common/cuda_is_ptr.hpp>
#include <utility>
#include <gtest/gtest.h>
#include <gridtools/common/cuda_util.hpp>
using gridtools::is_gpu_ptr;
using gridtools::cuda_util::cuda_malloc;
TEST(test_is_gpu_ptr, host_ptr_is_no_cuda_ptr) {
auto testee = std::unique_ptr<double>(new double);
EXPECT_FALSE(is_gpu_ptr(testee.get()));
EXPECT_EQ(cudaSuccess, cudaGetLastError());
}
TEST(test_is_gpu_ptr, cuda_ptr_is_cuda_ptr) {
auto testee = cuda_malloc<double>();
EXPECT_TRUE(is_gpu_ptr(testee.get()));
EXPECT_EQ(cudaSuccess, cudaGetLastError());
}
TEST(test_is_gpu_ptr, cuda_ptr_inner_region_are_cuda_ptr) {
auto testee = cuda_malloc<double[]>(2);
EXPECT_TRUE(is_gpu_ptr(testee.get() + 1));
EXPECT_EQ(cudaSuccess, cudaGetLastError());
}
|
