flatten_native_variants_to_staging()

void MayaFlux::Yantra::GpuDispatchCore::flatten_native_variants_to_staging ( const std::vector< Kakshya::DataVariant > &  variants, const DataStructureInfo &  structure_info  )

protected

Flatten native-typed DataVariant channels into m_native_staging_bytes without any conversion.

Called by prepare_gpu_inputs when structure_info.original_type indicates a non-double native type and no explicit stage_native_bytes call has been made. Visits the variant's active alternative and memcpys bytes directly.

No-ops when channels is empty or the modality is structured.

Parameters

variants	Per-channel DataVariants from the container.
structure_info	Dimension/modality metadata.

Definition at line 534 of file GpuDispatchCore.cpp.

{
    m_native_staging_bytes.clear();
 
    if (variants.empty() || Kakshya::is_structured_modality(structure_info.modality))
        return;
 
    size_t total_bytes = 0;
    for (const auto& v : variants) {
        std::visit([&](const auto& vec) {
            total_bytes += vec.size() * sizeof(typename std::decay_t<decltype(vec)>::value_type);
        },
            v);
    }
 
    m_native_staging_bytes.reserve(total_bytes);
 
    for (const auto& v : variants) {
        std::visit([&](const auto& vec) {
            using T = typename std::decay_t<decltype(vec)>::value_type;
            const auto* bytes = reinterpret_cast<const uint8_t*>(vec.data());
            m_native_staging_bytes.insert(
                m_native_staging_bytes.end(),
                bytes,
                bytes + vec.size() * sizeof(T));
        },
            v);
    }
}

References MayaFlux::Kakshya::is_structured_modality(), m_native_staging_bytes, and MayaFlux::Yantra::DataStructureInfo::modality.

Referenced by MayaFlux::Yantra::GpuExecutionContext< InputType, OutputType >::extract_inputs().

Here is the call graph for this function:

Here is the caller graph for this function: