Skip to content
Snippets Groups Projects

Extend Support for CUDA and HIP kernel invocations

Merged Frederik Hennig requested to merge fhennig/cuda-invoke into master
Compare and Show latest version
1 file
+ 2
2
Preferences
Compare changes
#include "CudaKernels.hpp"
#include <cuda/cuda_runtime.h>
#include <cuda_runtime.h>
#include <experimental/mdspan>
#include <random>
@@ -77,7 +77,7 @@ int main(void)
/* Linear3D Automatic */
cudaStream_t stream;
checkCudaError(cudaStreamCreate(&stream));
gen::linear3d::scaleKernel(dst, src, stream);
gen::linear3d_automatic::scaleKernel(dst, src, stream);
checkCudaError(cudaStreamSynchronize(stream)); });
check([&]()