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
|
/*
* 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_runtime.hpp>
#include <gridtools/storage/builder.hpp>
#include <gridtools/storage/gpu.hpp>
int main() {
int result_value = 42;
auto builder = gridtools::storage::builder<gridtools::storage::gpu>.type<int>().dimensions(2);
auto my_storage = builder();
my_storage->host_view()(0) = result_value;
auto dev_ptr = my_storage->get_target_ptr();
int host_buffer;
cudaMemcpy(&host_buffer, dev_ptr, sizeof(int), cudaMemcpyDeviceToHost);
if (host_buffer == result_value)
exit(0);
else
exit(1);
}
|
