rocm_parallelize_nested_map.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/structured_control_flow/map.h"
#include "sdfg/structured_control_flow/sequence.h"
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/transformations/transformation.h"
 
namespace sdfg {
namespace transformations {

class ROCMParallelizeNestedMap : public Transformation {
    structured_control_flow::StructuredLoop& loop_;
    size_t block_size_;
 
public:
    ROCMParallelizeNestedMap(structured_control_flow::StructuredLoop& loop, size_t block_size);
 
    virtual std::string name() const override;
 
    virtual bool
    can_be_applied(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
 
    virtual void apply(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
 
    virtual void to_json(nlohmann::json& j) const override;
 
    static ROCMParallelizeNestedMap from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);
};
 
 
} // namespace transformations
} // namespace sdfg