collect_gpu_outputs()

template<ComputeData InputType = std::vector<Kakshya::DataVariant>, ComputeData OutputType = InputType>

virtual output_type MayaFlux::Yantra::GpuExecutionContext< InputType, OutputType >::collect_gpu_outputs ( const GpuChannelResult &  raw, const std::vector< std::vector< double > > &  channels, const DataStructureInfo &  structure_info  )

inlineprotectedvirtual

Reconstruct Datum<OutputType> from a GpuChannelResult.

Reads the primary float readback into per-channel doubles and delegates to OperationHelper::reconstruct_from_double. Aux entries are forwarded into result.metadata keyed as "gpu_output_N".

Override to perform custom readback interpretation, e.g. when the output buffer layout does not match the input channel structure.

Parameters

raw	Readback produced by dispatch_core or dispatch_core_chained.
channels	Double channels extracted from the input Datum.
structure_info	Dimension/modality metadata.

Reimplemented in MayaFlux::Yantra::TextureExecutionContext.

Definition at line 116 of file GpuExecutionContext.hpp.

    {
        output_type result;
 
        const size_t total_ch_elements = std::accumulate(
            channels.begin(), channels.end(), size_t { 0 },
            [](size_t s, const auto& ch) { return s + ch.size(); });
 
        if (!raw.primary.empty() && !channels.empty()
            && raw.primary.size() >= total_ch_elements) {
            size_t offset = 0;
            std::vector<std::vector<double>> result_ch(channels.size());
            for (size_t c = 0; c < channels.size(); ++c) {
                result_ch[c].resize(channels[c].size());
                for (size_t i = 0; i < channels[c].size(); ++i)
                    result_ch[c][i] = static_cast<double>(raw.primary[offset++]);
            }
            result = Datum<OutputType>(
                OperationHelper::reconstruct_from_double<OutputType>(result_ch, structure_info));
        }
 
        for (const auto& [idx, bytes] : raw.aux)
            result.metadata["gpu_output_" + std::to_string(idx)] = bytes;
 
        return result;
    }

References MayaFlux::Yantra::GpuChannelResult::aux, MayaFlux::Yantra::Datum< T >::metadata, MayaFlux::Yantra::GpuChannelResult::primary, and size.