netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Temporarily remove test added by cl/672721411.

Browse Source 
PiperOrigin-RevId: 700746834
This commit is contained in:
Jamie Liu
2024-11-27 10:46:54 -08:00
committed by gVisor bot
parent bc18e81062
commit 78b9e4e0ae
1 changed files with 0 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files
images/gpu/cuda-tests/cuda_malloc.cu
-106
View File
@@ -29,108 +29,6 @@ __global__ void addKernel(std::uint32_t* data) {
data[index] += static_cast<std::uint32_t>(index);
}
void TestMallocHostReadWrite(int device) {
constexpr size_t kNumBlocks = 32;
constexpr size_t kNumThreads = 64;
constexpr size_t kNumElems = kNumBlocks * kNumThreads;
constexpr size_t kNumBytes = kNumElems * sizeof(std::uint32_t);
std::uint32_t* cpu_data = nullptr;
CHECK_CUDA(cudaMallocHost(&cpu_data, kNumBytes, cudaHostAllocWriteCombined));
std::uint32_t* gpu_data = nullptr;
CHECK_CUDA(cudaMalloc(&gpu_data, kNumBytes));
// Initialize all elements in the host array with a random value.
std::random_device rd;
const std::uint32_t init_val =
std::uniform_int_distribution<std::uint32_t>()(rd);
for (size_t i = 0; i < kNumElems; i++) {
cpu_data[i] = init_val;
}
// Write the host array's contents to a temporary file.
char filename[] = "/tmp/cudaMallocHostTest.XXXXXX";
int fd = mkstemp(filename);
if (fd < 0) {
err(1, "mkstemp");
}
size_t done = 0;
while (done < kNumBytes) {
ssize_t n = write(fd, reinterpret_cast<char*>(cpu_data) + done,
kNumBytes - done);
if (n >= 0) {
done += n;
} else if (n < 0 && errno != EINTR) {
err(1, "write");
}
}
// Copy the array to the device, mutate it there, and copy it back.
CHECK_CUDA(cudaMemcpy(gpu_data, cpu_data, kNumBytes, cudaMemcpyHostToDevice));
addKernel<<<kNumBlocks, kNumThreads>>>(gpu_data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
CHECK_CUDA(cudaMemcpy(cpu_data, gpu_data, kNumBytes, cudaMemcpyDeviceToHost));
// Check that the array has the expected result.
for (size_t i = 0; i < kNumElems; i++) {
std::uint32_t want = init_val + static_cast<std::uint32_t>(i);
if (cpu_data[i] != want) {
std::cout << "cpu_data[" << i << "]: got " << cpu_data[i] << ", wanted "
<< want << " = " << init_val << " + " << i << std::endl;
abort();
}
}
// Read the array's original contents back from the temporary file.
if (lseek(fd, 0, SEEK_SET) < 0) {
err(1, "lseek");
}
done = 0;
while (done < kNumBytes) {
ssize_t n = read(fd, reinterpret_cast<char*>(cpu_data) + done,
kNumBytes - done);
if (n > 0) {
done += n;
} else if (n == 0) {
errx(1, "read: unexpected EOF after %zu bytes", done);
} else if (n < 0 && errno != EINTR) {
err(1, "read");
}
}
// Check that the array matches what we originally wrote.
for (size_t i = 0; i < kNumElems; i++) {
std::uint32_t want = init_val;
if (cpu_data[i] != want) {
std::cout << "cpu_data[" << i << "]: got " << cpu_data[i] << ", wanted "
<< want << " = " << init_val << " + " << i << std::endl;
abort();
}
}
// Mutate the array on the device again.
CHECK_CUDA(cudaMemcpy(gpu_data, cpu_data, kNumBytes, cudaMemcpyHostToDevice));
addKernel<<<kNumBlocks, kNumThreads>>>(gpu_data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
CHECK_CUDA(cudaMemcpy(cpu_data, gpu_data, kNumBytes, cudaMemcpyDeviceToHost));
// Check that the array has the expected result again.
for (size_t i = 0; i < kNumElems; i++) {
std::uint32_t want = init_val + static_cast<std::uint32_t>(i);
if (cpu_data[i] != want) {
std::cout << "cpu_data[" << i << "]: got " << cpu_data[i] << ", wanted "
<< want << " = " << init_val << " + " << i << std::endl;
abort();
}
}
close(fd);
CHECK_CUDA(cudaFreeHost(cpu_data));
CHECK_CUDA(cudaFree(gpu_data));
}
void TestMallocManagedRoundTrip(int device, unsigned int malloc_flags,
bool prefetch) {
constexpr size_t kNumBlocks = 32;
@@ -274,10 +172,6 @@ int main() {
int device;
CHECK_CUDA(cudaGetDevice(&device));
std::cout << "Testing read/write syscalls on cudaMallocHost memory"
<< std::endl;
TestMallocHostReadWrite(device);
std::cout << "Testing cudaMallocManaged(flags=cudaMemAttachGlobal)"
<< std::endl;
TestMallocManagedRoundTrip(device, cudaMemAttachGlobal, false);