@@ -74,12 +74,12 @@ int main() {
74
74
LibreCUstream stream{};
75
75
CUDA_CHECK (libreCuStreamCreate (&stream, 0 ));
76
76
77
- void *float_dst_compute_va {};
78
- void *float_dst_dma_va {};
79
- CUDA_CHECK (libreCuMemAlloc (&float_dst_compute_va , sizeof (float ), true ));
80
- CUDA_CHECK (libreCuMemAlloc (&float_dst_dma_va , sizeof (float ), true ));
81
- *(float *) float_dst_compute_va = 0 .0f ;
82
- *(float *) float_dst_dma_va = 0 .0f ;
77
+ void *float_dst_va {};
78
+ void *float_src_va {};
79
+ CUDA_CHECK (libreCuMemAlloc (&float_dst_va , sizeof (float ), true ));
80
+ CUDA_CHECK (libreCuMemAlloc (&float_src_va , sizeof (float ), true ));
81
+ *(float *) float_dst_va = 0 .0f ;
82
+ *(float *) float_src_va = 1 .0f ;
83
83
84
84
LibreCUEvent start{}, end{};
85
85
CUDA_CHECK (libreCuEventCreate (&start, 0 ));
@@ -88,7 +88,7 @@ int main() {
88
88
CUDA_CHECK (libreCuEventRecord (start, stream));
89
89
{
90
90
void *params[] = {
91
- &float_dst_compute_va , &float_dst_dma_va
91
+ &float_dst_va , &float_src_va
92
92
};
93
93
CUDA_CHECK (
94
94
libreCuLaunchKernel (func,
@@ -111,12 +111,11 @@ int main() {
111
111
112
112
CUDA_CHECK (libreCuStreamAwait (stream));
113
113
114
- std::cout << " Dst compute value (post exec): " << *(float *) (float_dst_compute_va) << std::endl;
115
- std::cout << " Dst dma value (post exec): " << *(float *) (float_dst_dma_va) << std::endl;
114
+ std::cout << " Dst value (post exec): " << *(float *) (float_dst_va) << std::endl;
116
115
117
116
// free memory
118
- CUDA_CHECK (libreCuMemFree (float_dst_compute_va ));
119
- CUDA_CHECK (libreCuMemFree (float_dst_dma_va ));
117
+ CUDA_CHECK (libreCuMemFree (float_dst_va ));
118
+ CUDA_CHECK (libreCuMemFree (float_src_va ));
120
119
121
120
// destroy stream
122
121
CUDA_CHECK (libreCuStreamDestroy (stream));
0 commit comments