GpuSorter.hpp

Go to the documentation of this file.

#pragma once
 
#include "MayaFlux/Yantra/Executors/GpuExecutionContext.hpp"
#include "UniversalSorter.hpp"
 
namespace MayaFlux::Yantra {
 
/**
 * @class GpuSorter
 * @brief Concrete UniversalSorter that dispatches entirely via a
 *        GpuExecutionContext. CPU path is a hard error.
 *
 * @tparam InputType  ComputeData type accepted.
 * @tparam OutputType ComputeData type produced.
 */
template <ComputeData InputType = std::vector<Kakshya::DataVariant>,
    ComputeData OutputType = InputType>
class MAYAFLUX_API GpuSorter : public UniversalSorter<InputType, OutputType> {
public:
    using input_type = Datum<InputType>;
    using output_type = Datum<OutputType>;
 
    /**
     * @brief Construct with a configured GpuExecutionContext.
     * @param executor Configured executor. Must not be null.
     */
    explicit GpuSorter(
        std::shared_ptr<GpuExecutionContext<InputType, OutputType>> executor)
    {
        assert(executor && "GpuSorter: executor must not be null");
        m_executor = executor;
        this->set_gpu_backend(std::move(executor));
    }
 
    /**
     * @brief Returns the attached GpuExecutionContext for further configuration.
     */
    [[nodiscard]] std::shared_ptr<GpuExecutionContext<InputType, OutputType>>
    get_executor() const { return m_executor; }
 
    [[nodiscard]] std::string get_sorter_name() const override
    {
        return "GpuSorter";
    }
 
    [[nodiscard]] SortingType get_sorting_type() const override
    {
        return SortingType::CUSTOM;
    }
 
protected:
    output_type sort_implementation(const input_type&) override
    {
        error<std::runtime_error>(
            Journal::Component::Yantra,
            Journal::Context::BufferProcessing,
            std::source_location::current(),
            "GpuSorter: GPU unavailable and no CPU fallback provided");
    }
 
private:
    std::shared_ptr<GpuExecutionContext<InputType, OutputType>> m_executor;
};
 
} // namespace MayaFlux::Yantra