map_resources_scope.h

#include <cuda_runtime.h>
#include <vector>
#include "error.h"

namespace utils::cuda {
    class MapResourcesScope {
    public:
        MapResourcesScope(const std::vector<cudaGraphicsResource_t> &resources,
                          cudaStream_t stream = nullptr)
            : _resources(resources), _stream(stream) {
            if (!_resources.empty())
                THROW_IF_FAILED(
                    cudaGraphicsMapResources((int)_resources.size(), _resources.data(), _stream));
        }

        ~MapResourcesScope() {
            if (!_resources.empty())
                cudaGraphicsUnmapResources((int)_resources.size(), _resources.data(), _stream);
        }

    private:
        std::vector<cudaGraphicsResource_t> _resources;
        cudaStream_t _stream;
    };
} // namespace utils::cuda