Pcornat/raii-safe-cuda
1
0
Fork 0
Code Activity

refactor repo

Browse source
This commit is contained in:
Pcornat 2024-03-25 18:13:16 +01:00
commit 24a83dfa08
Signed by: Pcornat
GPG key ID: E0326CC678A00BDD
7 changed files with 57 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

21
safe_cuda/stream_related.h
View file

@ -1,21 +0,0 @@
#ifndef RAIISAFECUDA_STREAM_RELATED_H
#define RAIISAFECUDA_STREAM_RELATED_H
#include <memory>
#include <cuda_runtime_api.h>
#include <variant>
namespace safe_cuda {
template<typename T>
using returnType = std::variant<T, cudaError_t>;
using streamDestroyType = decltype(&cudaStreamDestroy);
/**
* \brief It tries to create a stream, putting it in a smart pointer with its correct destructor.
* \return smart pointer if creation is OK, else the CUDA error
*/
returnType<std::unique_ptr<CUstream_st, streamDestroyType>> create_stream() noexcept;
}
#endif //RAIISAFECUDA_STREAM_RELATED_H