26448855920

Committed 26 May 2026 12:46PM UTC coverage: 60.884% (+0.04%) from 60.848%

Build # 26448855920

Build Type

push

github

Committed by

web-flow

Commit Message

Merge pull request #721 from daisytuner/second-level-tiling

Make second level tiling possible

Coverage Stats

74 of 92 new or added lines in 2 files covered. (80.43%)

2 existing lines in 1 file now uncovered.

35101 of 57652 relevant lines covered (60.88%)

11087.7 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

24.66

/opt/src/transformations/replayer.cpp

#include <sdfg/transformations/einsum2dot.h>
#include <sdfg/transformations/einsum2gemm.h>
#include <sdfg/transformations/einsum_extend.h>
#include <sdfg/transformations/einsum_lift.h>
#include <sdfg/transformations/einsum_promotion.h>
#include <sdfg/transformations/in_local_storage.h>
#include <sdfg/transformations/loop_distribute.h>
#include <sdfg/transformations/loop_interchange.h>
#include <sdfg/transformations/loop_peeling.h>
#include <sdfg/transformations/loop_shift.h>
#include <sdfg/transformations/loop_skewing.h>
#include <sdfg/transformations/loop_split.h>
#include <sdfg/transformations/loop_tiling.h>
#include <sdfg/transformations/multi_level_tiling.h>
#include <sdfg/transformations/offloading/cuda_parallelize_nested_map.h>
#include <sdfg/transformations/offloading/cuda_transform.h>
#include <sdfg/transformations/offloading/gpu_condition_propagation.h>
#include <sdfg/transformations/offloading/gpu_loop_reordering.h>
#include <sdfg/transformations/offloading/gpu_tiling.h>
#include <sdfg/transformations/offloading/kernel_local_storage.h>
#include <sdfg/transformations/offloading/rocm_parallelize_nested_map.h>
#include <sdfg/transformations/offloading/rocm_transform.h>
#include <sdfg/transformations/omp_transform.h>
#include <sdfg/transformations/out_local_storage.h>
#include <sdfg/transformations/replayer.h>
#include <sdfg/transformations/tile_fusion.h>
#include <sdfg/transformations/vectorize_transform.h>

namespace sdfg {
namespace transformations {

void Replayer::replay(
    builder::StructuredSDFGBuilder& builder,
    analysis::AnalysisManager& analysis_manager,
    const nlohmann::json& transformation_data,
    bool skip_if_not_applicable,
    size_t loopnest_index
) {
    if (!transformation_data.is_array()) {
        throw std::runtime_error("Transformation data must be an array.");
    }

    for (const auto& desc : transformation_data) {
        auto transformation_name = desc["transformation_type"];

        if (transformation_name == "LoopTiling") {
            this->apply<transformations::LoopTiling>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "MultiLevelTiling") {
            this->apply<transformations::MultiLevelTiling>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "LoopDistribute") {
            this->apply<transformations::LoopDistribute>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "LoopInterchange") {
            this->apply<transformations::LoopInterchange>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "OutLocalStorage") {
            this->apply<transformations::OutLocalStorage>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "InLocalStorage") {
            this->apply<transformations::InLocalStorage>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "TileFusion") {
            this->apply<transformations::TileFusion>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "LoopSkewing") {
            this->apply<transformations::LoopSkewing>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "LoopShift") {
            this->apply<transformations::LoopShift>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "LoopSplit") {
            this->apply<transformations::LoopSplit>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "OMPTransform") {
            this->apply<transformations::OMPTransform>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "LoopPeeling") {
            this->apply<transformations::LoopPeeling>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "VectorizeTransform") {
            this->apply<transformations::VectorizeTransform>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "CUDATransform") {
            this->apply<cuda::CUDATransform>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "CUDAParallelizeNestedMap") {
            this->apply<
                transformations::CUDAParallelizeNestedMap>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "ROCMTransform") {
            this->apply<rocm::ROCMTransform>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "ROCMParallelizeNestedMap") {
            this->apply<
                transformations::ROCMParallelizeNestedMap>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "GPUConditionPropagation") {
            this->apply<
                transformations::GPUConditionPropagation>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "GPUTiling") {
            this->apply<transformations::GPUTiling>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "GPULoopReordering") {
            this->apply<transformations::GPULoopReordering>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "KernelLocalStorage") {
            this->apply<transformations::KernelLocalStorage>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "EinsumLift") {
            this->apply<transformations::EinsumLift>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "EinsumExtend") {
            this->apply<transformations::EinsumExtend>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "EinsumExpand") {
            this->apply<transformations::EinsumPromotion>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "Einsum2Dot") {
            this->apply<transformations::Einsum2Dot>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else if (transformation_name == "Einsum2Gemm") {
            this->apply<transformations::Einsum2Gemm>(builder, analysis_manager, desc, skip_if_not_applicable);
        } else {
            throw transformations::InvalidTransformationDescriptionException(
                "Unknown transformation: " + transformation_name.get<std::string>()
            );
        }

#ifndef NDEBUG
        std::cout << "Applied transformation: " << transformation_name << std::endl;
        builder.subject().validate();
#endif

        analysis_manager.invalidate_all();
    }
}


} // namespace transformations
} // namespace sdfg

1	#include <sdfg/transformations/einsum2dot.h>
2	#include <sdfg/transformations/einsum2gemm.h>
3	#include <sdfg/transformations/einsum_extend.h>
4	#include <sdfg/transformations/einsum_lift.h>
5	#include <sdfg/transformations/einsum_promotion.h>
6	#include <sdfg/transformations/in_local_storage.h>
7	#include <sdfg/transformations/loop_distribute.h>
8	#include <sdfg/transformations/loop_interchange.h>
9	#include <sdfg/transformations/loop_peeling.h>
10	#include <sdfg/transformations/loop_shift.h>
11	#include <sdfg/transformations/loop_skewing.h>
12	#include <sdfg/transformations/loop_split.h>
13	#include <sdfg/transformations/loop_tiling.h>
14	#include <sdfg/transformations/multi_level_tiling.h>
15	#include <sdfg/transformations/offloading/cuda_parallelize_nested_map.h>
16	#include <sdfg/transformations/offloading/cuda_transform.h>
17	#include <sdfg/transformations/offloading/gpu_condition_propagation.h>
18	#include <sdfg/transformations/offloading/gpu_loop_reordering.h>
19	#include <sdfg/transformations/offloading/gpu_tiling.h>
20	#include <sdfg/transformations/offloading/kernel_local_storage.h>
21	#include <sdfg/transformations/offloading/rocm_parallelize_nested_map.h>
22	#include <sdfg/transformations/offloading/rocm_transform.h>
23	#include <sdfg/transformations/omp_transform.h>
24	#include <sdfg/transformations/out_local_storage.h>
25	#include <sdfg/transformations/replayer.h>
26	#include <sdfg/transformations/tile_fusion.h>
27	#include <sdfg/transformations/vectorize_transform.h>
28
29	namespace sdfg {
30	namespace transformations {
31
32	void Replayer::replay(
33	builder::StructuredSDFGBuilder& builder,
34	analysis::AnalysisManager& analysis_manager,
35	const nlohmann::json& transformation_data,
36	bool skip_if_not_applicable,
37	size_t loopnest_index
38	) {	5✔
39	if (!transformation_data.is_array()) {	5✔
40	throw std::runtime_error("Transformation data must be an array.");	×
41	}	×
42
43	for (const auto& desc : transformation_data) {	9✔
44	auto transformation_name = desc["transformation_type"];	9✔
45
46	if (transformation_name == "LoopTiling") {	9✔
47	this->apply<transformations::LoopTiling>(builder, analysis_manager, desc, skip_if_not_applicable);	7✔
48	} else if (transformation_name == "MultiLevelTiling") {	7✔
NEW 49	this->apply<transformations::MultiLevelTiling>(builder, analysis_manager, desc, skip_if_not_applicable);	×
50	} else if (transformation_name == "LoopDistribute") {	2✔
51	this->apply<transformations::LoopDistribute>(builder, analysis_manager, desc, skip_if_not_applicable);	×
52	} else if (transformation_name == "LoopInterchange") {	2✔
53	this->apply<transformations::LoopInterchange>(builder, analysis_manager, desc, skip_if_not_applicable);	2✔
54	} else if (transformation_name == "OutLocalStorage") {	2✔
55	this->apply<transformations::OutLocalStorage>(builder, analysis_manager, desc, skip_if_not_applicable);	×
56	} else if (transformation_name == "InLocalStorage") {	×
57	this->apply<transformations::InLocalStorage>(builder, analysis_manager, desc, skip_if_not_applicable);	×
58	} else if (transformation_name == "TileFusion") {	×
59	this->apply<transformations::TileFusion>(builder, analysis_manager, desc, skip_if_not_applicable);	×
60	} else if (transformation_name == "LoopSkewing") {	×
61	this->apply<transformations::LoopSkewing>(builder, analysis_manager, desc, skip_if_not_applicable);	×
62	} else if (transformation_name == "LoopShift") {	×
63	this->apply<transformations::LoopShift>(builder, analysis_manager, desc, skip_if_not_applicable);	×
64	} else if (transformation_name == "LoopSplit") {	×
65	this->apply<transformations::LoopSplit>(builder, analysis_manager, desc, skip_if_not_applicable);	×
66	} else if (transformation_name == "OMPTransform") {	×
67	this->apply<transformations::OMPTransform>(builder, analysis_manager, desc, skip_if_not_applicable);	×
68	} else if (transformation_name == "LoopPeeling") {	×
69	this->apply<transformations::LoopPeeling>(builder, analysis_manager, desc, skip_if_not_applicable);	×
70	} else if (transformation_name == "VectorizeTransform") {	×
71	this->apply<transformations::VectorizeTransform>(builder, analysis_manager, desc, skip_if_not_applicable);	×
72	} else if (transformation_name == "CUDATransform") {	×
73	this->apply<cuda::CUDATransform>(builder, analysis_manager, desc, skip_if_not_applicable);	×
74	} else if (transformation_name == "CUDAParallelizeNestedMap") {	×
75	this->apply<	×
76	transformations::CUDAParallelizeNestedMap>(builder, analysis_manager, desc, skip_if_not_applicable);	×
77	} else if (transformation_name == "ROCMTransform") {	×
78	this->apply<rocm::ROCMTransform>(builder, analysis_manager, desc, skip_if_not_applicable);	×
79	} else if (transformation_name == "ROCMParallelizeNestedMap") {	×
80	this->apply<	×
81	transformations::ROCMParallelizeNestedMap>(builder, analysis_manager, desc, skip_if_not_applicable);	×
82	} else if (transformation_name == "GPUConditionPropagation") {	×
83	this->apply<	×
84	transformations::GPUConditionPropagation>(builder, analysis_manager, desc, skip_if_not_applicable);	×
85	} else if (transformation_name == "GPUTiling") {	×
86	this->apply<transformations::GPUTiling>(builder, analysis_manager, desc, skip_if_not_applicable);	×
87	} else if (transformation_name == "GPULoopReordering") {	×
88	this->apply<transformations::GPULoopReordering>(builder, analysis_manager, desc, skip_if_not_applicable);	×
89	} else if (transformation_name == "KernelLocalStorage") {	×
90	this->apply<transformations::KernelLocalStorage>(builder, analysis_manager, desc, skip_if_not_applicable);	×
91	} else if (transformation_name == "EinsumLift") {	×
92	this->apply<transformations::EinsumLift>(builder, analysis_manager, desc, skip_if_not_applicable);	×
93	} else if (transformation_name == "EinsumExtend") {	×
94	this->apply<transformations::EinsumExtend>(builder, analysis_manager, desc, skip_if_not_applicable);	×
95	} else if (transformation_name == "EinsumExpand") {	×
96	this->apply<transformations::EinsumPromotion>(builder, analysis_manager, desc, skip_if_not_applicable);	×
97	} else if (transformation_name == "Einsum2Dot") {	×
98	this->apply<transformations::Einsum2Dot>(builder, analysis_manager, desc, skip_if_not_applicable);	×
99	} else if (transformation_name == "Einsum2Gemm") {	×
100	this->apply<transformations::Einsum2Gemm>(builder, analysis_manager, desc, skip_if_not_applicable);	×
101	} else {	×
102	throw transformations::InvalidTransformationDescriptionException(	×
103	"Unknown transformation: " + transformation_name.get<std::string>()	×
104	);	×
105	}	×
106
107	#ifndef NDEBUG	9✔
108	std::cout << "Applied transformation: " << transformation_name << std::endl;	9✔
109	builder.subject().validate();	9✔
110	#endif	9✔
111
112	analysis_manager.invalidate_all();	9✔
113	}	9✔
114	}	5✔
115
116
117	} // namespace transformations
118	} // namespace sdfg

daisytuner / docc / 26448855920

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous