23639739465

Committed 27 Mar 2026 09:27AM UTC coverage: 64.981% (+0.1%) from 64.845%

Build # 23639739465

Build Type

Pull #605

github

Committed by

web-flow

Commit Message

Merge 47e169e74 into c252af595

Pull Request Pull Request #605: Move einsum support

Coverage Stats

1385 of 2014 new or added lines in 15 files covered. (68.77%)

206 existing lines in 1 file now uncovered.

28290 of 43536 relevant lines covered (64.98%)

394.35 hits per line

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

26.98

/opt/src/transformations/replayer.cpp

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

daisytuner / docc / 23639739465

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