blob: 77479f85dc5cc3f82e9bb077b48ee259c25b071e (
plain)
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
|
//===-- Test for parallel GPU malloc interface ----------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "test/IntegrationTest/test.h"
#include "src/__support/GPU/utils.h"
#include "src/stdlib/free.h"
#include "src/stdlib/malloc.h"
using namespace LIBC_NAMESPACE;
static inline void use(uint8_t *ptr, uint32_t size) {
EXPECT_NE(ptr, nullptr);
for (int i = 0; i < size; ++i)
ptr[i] = uint8_t(i + gpu::get_thread_id());
// Try to detect if some other thread manages to clobber our memory.
for (int i = 0; i < size; ++i)
EXPECT_EQ(ptr[i], uint8_t(i + gpu::get_thread_id()));
}
TEST_MAIN(int, char **, char **) {
void *ptrs[256];
for (int i = 0; i < 256; ++i)
ptrs[i] = malloc(gpu::get_lane_id() % 2 ? 16 : 32);
for (int i = 0; i < 256; ++i)
use(reinterpret_cast<uint8_t *>(ptrs[i]), gpu::get_lane_id() % 2 ? 16 : 32);
for (int i = 0; i < 256; ++i)
free(ptrs[i]);
return 0;
}
|
