fusion_registry.h

Go to the documentation of this file.

#pragma once
 
/*
 * ADVANCED API — no source-compatibility promise. The types here (CompressionDAG,
 * DAGNode, BufferInfo, the fusion planner/registry) are pipeline internals exposed
 * for advanced/experimental use; they may change or be removed in any release.
 * Most users only need <fzgpumodules.h> / pipeline/compressor.h. See the API tiers
 * in docs/api_reference.md.
 */
 
#include "backend/types.h"
#include "stage/fusion.h"
#include <cstddef>
#include <string>
#include <vector>
 
namespace fz {
 
class Stage;
class MemoryPool;
 
struct FusedSideOutput {
    Stage* producer;       
    int    output_index;   
    void*  d_ptr;          
    size_t capacity;       
    size_t size;           
    FusedAuxOutputDecl declaration; 
};
 
struct FusedSideInput {
    Stage* consumer;       
    int    input_index;    
    const void* d_ptr;     
    size_t size;           
};
 
struct FusedRunContext {
    const std::vector<Stage*>* stages;   
    const void* d_input;                 
    size_t      input_bytes;             
    void*       d_output;                
    size_t      output_capacity = 0;     
    MemoryPool* pool;                    
    fz::stream_t stream;                 
    std::vector<FusedSideOutput>* side_outputs = nullptr;
    const std::vector<FusedSideInput>* side_inputs = nullptr;
    std::string* execution_path = nullptr;
};
 
struct FusedImpl {
    const char* name;
    bool auto_enabled;
    bool   (*matches)(const std::vector<Stage*>& group);
    size_t (*run)(const FusedRunContext& ctx);
};
 
const FusedImpl* findFusedImpl(
    const std::vector<Stage*>& group, bool include_experimental = false);
 
} // namespace fz