rpc_responses.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <nlohmann/json.hpp>
#include "sdfg/analysis/loop_analysis.h"
#include "sdfg/structured_sdfg.h"
 
namespace sdfg::passes::rpc {
 
struct RpcOptimizationMetadata {
    std::optional<std::string> region_id;
    std::optional<nlohmann::json> expected_performance;
    std::optional<double> vector_distance;
};

struct RpcLocalReplayRecipe {
    nlohmann::json sequence;
};
 
struct RpcSdfgResult {
    std::unique_ptr<sdfg::StructuredSDFG> sdfg = nullptr;
};

struct RpcRegionResult {
    std::optional<int64_t> element_id;
    std::optional<RpcLocalReplayRecipe> local_replay;
    RpcOptimizationMetadata metadata;
};
 
struct RpcOptResponse {
    std::optional<RpcSdfgResult> sdfg_result;
    std::optional<RpcLocalReplayRecipe> local_replay;
    std::vector<RpcRegionResult> results;
    std::optional<std::string> session_id;
    std::optional<std::string> error;
};
 
struct RpcOptRequest {
    StructuredSDFG& sdfg;
    std::string category;
    std::string target;
    analysis::LoopInfo loop_info;
    bool enable_fusion = true;
    bool normalize = true;
    bool schedule_loops = true;
    std::optional<std::string> session_id;
    std::optional<std::string> collection;
    bool use_identity = false;
    std::optional<double> distance_threshold;
};
 
} // namespace sdfg::passes::rpc