24180171226

Committed 09 Apr 2026 08:21AM UTC coverage: 64.38% (-0.6%) from 64.986%

Build # 24180171226

Build Type

push

github

Committed by

web-flow

Commit Message

Extract Implicit Data Transfers from GPU BLAS Libnodes (#662)

* GPU blas data transfer extraction draft

* reuse original container type

* fix input containers for gemm nodes

Coverage Stats

215 of 320 new or added lines in 11 files covered. (67.19%)

29659 of 46069 relevant lines covered (64.38%)

588.86 hits per line

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

60.0

/opt/src/passes/offloading/cublas_offloading_expansion_pass.cpp

#include "sdfg/passes/offloading/cublas_offloading_expansion_pass.h"

#include "sdfg/data_flow/library_nodes/math/blas/blas_node.h"
#include "sdfg/transformations/offloading/cublas_offloading_expansion.h"

namespace sdfg {
namespace cuda {

CublasBLASOffloadingExpansionVisitor::CublasBLASOffloadingExpansionVisitor(
    builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager
)
    : visitor::NonStoppingStructuredSDFGVisitor(builder, analysis_manager) {}

bool CublasBLASOffloadingExpansionVisitor::visit() { return visitor::NonStoppingStructuredSDFGVisitor::visit(); }

bool CublasBLASOffloadingExpansionVisitor::accept(structured_control_flow::Block& block) {
    auto& dataflow = block.dataflow();
    for (auto lib_node : dataflow.library_nodes()) {
        if (auto* blas_node = dynamic_cast<math::blas::BLASNode*>(lib_node)) {
            CUBLASOffloadingExpansion expansion(*blas_node);
            if (expansion.can_be_applied(builder_, analysis_manager_)) {
                expansion.apply(builder_, analysis_manager_);
                return true;
            }
        }
    }
    return false;
}

} // namespace cuda
} // namespace sdfg

1	#include "sdfg/passes/offloading/cublas_offloading_expansion_pass.h"
2
3	#include "sdfg/data_flow/library_nodes/math/blas/blas_node.h"
4	#include "sdfg/transformations/offloading/cublas_offloading_expansion.h"
5
6	namespace sdfg {
7	namespace cuda {
8
9	CublasBLASOffloadingExpansionVisitor::CublasBLASOffloadingExpansionVisitor(
10	builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager
11	)
12	: visitor::NonStoppingStructuredSDFGVisitor(builder, analysis_manager) {}	3✔
13
14	bool CublasBLASOffloadingExpansionVisitor::visit() { return visitor::NonStoppingStructuredSDFGVisitor::visit(); }	3✔
15
16	bool CublasBLASOffloadingExpansionVisitor::accept(structured_control_flow::Block& block) {	12✔
17	auto& dataflow = block.dataflow();	12✔
18	for (auto lib_node : dataflow.library_nodes()) {	12✔
19	if (auto* blas_node = dynamic_cast<math::blas::BLASNode*>(lib_node)) {	8✔
NEW 20	CUBLASOffloadingExpansion expansion(*blas_node);	×
NEW 21	if (expansion.can_be_applied(builder_, analysis_manager_)) {	×
NEW 22	expansion.apply(builder_, analysis_manager_);	×
NEW 23	return true;	×
NEW 24	}	×
NEW 25	}	×
26	}	8✔
27	return false;	12✔
28	}	12✔
29
30	} // namespace cuda
31	} // namespace sdfg

daisytuner / docc / 24180171226

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