fusion_planner.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 "stage/fusion.h"
#include <cstdint>
#include <string>
#include <vector>
 
namespace fz {
 
class Stage;
class CompressionDAG;
 
struct FusionGeometry {
    uint32_t block_size = 0;         
    uint32_t selector_tile_size = 0; 
    uint32_t coder_unit_size = 0;    
 
    bool hasTileSelector() const { return selector_tile_size != 0; }
};
 
enum class FusionCompatibility : uint8_t {
    Compatible = 0,
    UnfusableStage,
    InvalidTileGeometry,
    TileAfterRegionLocal,
    MultipleTileSelectors,
    TileInteriorStageUnsupported,
    StandardBlockMismatch,
    TileCoderUnitMismatch,
};
 
FusionCompatibility extendFusionGeometry(
    FusionGeometry& geometry, const FusionSpec& next);
 
const char* fusionCompatibilityName(FusionCompatibility result);
 
struct FusionGroup {
    std::vector<Stage*>      stages;       
    std::vector<std::string> stage_names;  
    uint32_t                 block_size = 0;   
    uint32_t                 selector_tile_size = 0; 
    uint32_t                 coder_unit_size = 0;    
    bool                     has_tile_selector = false;
    bool                     has_coder  = false;
};
 
std::vector<FusionGroup> planFusionGroups(const CompressionDAG& dag);
 
} // namespace fz