init

safe_cuda/stream_related.cpp

@@ -0,0 +1,13 @@
+#include "stream_related.h"
+
+namespace safe_cuda {
+    returnType<std::unique_ptr<CUstream_st, streamDestroyType>> create_stream() noexcept {
+        cudaStream_t stream = nullptr;
+        const cudaError_t error = cudaStreamCreate(&stream);
+        if (error != cudaSuccess) {
+            return error;
+        }
+        return std::unique_ptr<CUstream_st, streamDestroyType>{ stream, cudaStreamDestroy };
+    }
+
+}

safe_cuda/stream_related.h

@@ -0,0 +1,21 @@
+#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