Skip to content

Commit

Permalink
fix meaningless variable naming copied from other test
Browse files Browse the repository at this point in the history
  • Loading branch information
mikex86 committed Sep 6, 2024
1 parent cb2606a commit 4525094
Showing 1 changed file with 10 additions and 11 deletions.
21 changes: 10 additions & 11 deletions tests/stream_events/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -74,12 +74,12 @@ int main() {
LibreCUstream stream{};
CUDA_CHECK(libreCuStreamCreate(&stream, 0));

void *float_dst_compute_va{};
void *float_dst_dma_va{};
CUDA_CHECK(libreCuMemAlloc(&float_dst_compute_va, sizeof(float), true));
CUDA_CHECK(libreCuMemAlloc(&float_dst_dma_va, sizeof(float), true));
*(float *) float_dst_compute_va = 0.0f;
*(float *) float_dst_dma_va = 0.0f;
void *float_dst_va{};
void *float_src_va{};
CUDA_CHECK(libreCuMemAlloc(&float_dst_va, sizeof(float), true));
CUDA_CHECK(libreCuMemAlloc(&float_src_va, sizeof(float), true));
*(float *) float_dst_va = 0.0f;
*(float *) float_src_va = 1.0f;

LibreCUEvent start{}, end{};
CUDA_CHECK(libreCuEventCreate(&start, 0));
Expand All @@ -88,7 +88,7 @@ int main() {
CUDA_CHECK(libreCuEventRecord(start, stream));
{
void *params[] = {
&float_dst_compute_va, &float_dst_dma_va
&float_dst_va, &float_src_va
};
CUDA_CHECK(
libreCuLaunchKernel(func,
Expand All @@ -111,12 +111,11 @@ int main() {

CUDA_CHECK(libreCuStreamAwait(stream));

std::cout << "Dst compute value (post exec): " << *(float *) (float_dst_compute_va) << std::endl;
std::cout << "Dst dma value (post exec): " << *(float *) (float_dst_dma_va) << std::endl;
std::cout << "Dst value (post exec): " << *(float *) (float_dst_va) << std::endl;

// free memory
CUDA_CHECK(libreCuMemFree(float_dst_compute_va));
CUDA_CHECK(libreCuMemFree(float_dst_dma_va));
CUDA_CHECK(libreCuMemFree(float_dst_va));
CUDA_CHECK(libreCuMemFree(float_src_va));

// destroy stream
CUDA_CHECK(libreCuStreamDestroy(stream));
Expand Down

0 comments on commit 4525094

Please sign in to comment.