26831349290

Committed 02 Jun 2026 03:50PM UTC coverage: 61.29% (-0.01%) from 61.302%

Build # 26831349290

Build Type

Pull #725

github

Committed by

web-flow

Commit Message

Merge a7e2175c0 into 887730e20

Pull Request Pull Request #725: Tensor node backport

Coverage Stats

932 of 1642 new or added lines in 52 files covered. (56.76%)

92 existing lines in 33 files now uncovered.

35584 of 58058 relevant lines covered (61.29%)

11020.18 hits per line

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

0.0

/sdfg/src/data_flow/library_nodes/math/tensor/elementwise_ops/fill_node.cpp

#include "sdfg/data_flow/library_nodes/math/tensor/elementwise_ops/fill_node.h"

#include "sdfg/analysis/analysis.h"
#include "sdfg/analysis/scope_analysis.h"
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/data_flow/access_node.h"
#include "sdfg/data_flow/library_nodes/math/math_node.h"
#include "sdfg/types/type.h"

namespace sdfg {
namespace math {
namespace tensor {

FillNode::FillNode(
    size_t element_id,
    const DebugInfo& debug_info,
    const graph::Vertex vertex,
    data_flow::DataFlowGraph& parent,
    const std::vector<symbolic::Expression>& shape,
    QuantizationType quantization,
    const data_flow::ImplementationType& impl_type
)
    : ElementWiseDataflowTensorNode(
          element_id, debug_info, vertex, parent, LibraryNodeType_Fill, shape, "Y", {"X"}, quantization, impl_type
      ) {}

ElementWiseDataflowTensorNode::ElementOutput FillNode::expand_operation_dataflow(
    builder::StructuredSDFGBuilder& builder,
    analysis::AnalysisManager& analysis_manager,
    Block& block,
    std::vector<ElementInput>& needed_inputs,
    types::PrimitiveType expected_type
) {
    auto& tasklet = builder.add_tasklet(block, data_flow::TaskletCode::assign, "_out", {"_in"});
    auto& input = needed_inputs.at(0);
    input.consumer = &tasklet;
    input.input_conn_index = 0;

    return {.producer = &tasklet, .output_conn_index = 0, .type = input.required_type};
}

std::unique_ptr<data_flow::DataFlowNode> FillNode::
    clone(size_t element_id, const graph::Vertex vertex, data_flow::DataFlowGraph& parent) const {
    return std::unique_ptr<data_flow::DataFlowNode>(new FillNode(
        element_id, this->debug_info(), vertex, parent, this->shape_, fixed_quantization_, implementation_type_
    ));
}

void FillNode::validate_non_tensor_inputs(const data_flow::DataFlowGraph& graph) const {
    auto* fill_value_edge = graph.in_edge_for_connector(*this, inputs_.at(1));
    // Validate that the input is a scalar
    if (fill_value_edge && fill_value_edge->base_type().type_id() != types::TypeID::Scalar) {
        throw InvalidSDFGException(
            "FillNode: Input memlet must be of scalar type. Found type: " + fill_value_edge->base_type().print()
        );
    }
}

} // namespace tensor
} // namespace math
} // namespace sdfg

1	#include "sdfg/data_flow/library_nodes/math/tensor/elementwise_ops/fill_node.h"
2
3	#include "sdfg/analysis/analysis.h"
4	#include "sdfg/analysis/scope_analysis.h"
5	#include "sdfg/builder/structured_sdfg_builder.h"
6	#include "sdfg/data_flow/access_node.h"
7	#include "sdfg/data_flow/library_nodes/math/math_node.h"
8	#include "sdfg/types/type.h"
9
10	namespace sdfg {
11	namespace math {
12	namespace tensor {
13
14	FillNode::FillNode(
15	size_t element_id,
16	const DebugInfo& debug_info,
17	const graph::Vertex vertex,
18	data_flow::DataFlowGraph& parent,
19	const std::vector<symbolic::Expression>& shape,
20	QuantizationType quantization,
21	const data_flow::ImplementationType& impl_type
22	)
NEW 23	: ElementWiseDataflowTensorNode(	×
NEW 24	element_id, debug_info, vertex, parent, LibraryNodeType_Fill, shape, "Y", {"X"}, quantization, impl_type	×
NEW 25	) {}	×
26
27	ElementWiseDataflowTensorNode::ElementOutput FillNode::expand_operation_dataflow(
28	builder::StructuredSDFGBuilder& builder,
29	analysis::AnalysisManager& analysis_manager,
30	Block& block,
31	std::vector<ElementInput>& needed_inputs,
32	types::PrimitiveType expected_type
UNCOV 33	) {	×
NEW 34	auto& tasklet = builder.add_tasklet(block, data_flow::TaskletCode::assign, "_out", {"_in"});	×
NEW 35	auto& input = needed_inputs.at(0);	×
NEW 36	input.consumer = &tasklet;	×
NEW 37	input.input_conn_index = 0;	×
38
NEW 39	return {.producer = &tasklet, .output_conn_index = 0, .type = input.required_type};	×
40	}	×
41
42	std::unique_ptr<data_flow::DataFlowNode> FillNode::
43	clone(size_t element_id, const graph::Vertex vertex, data_flow::DataFlowGraph& parent) const {	×
NEW 44	return std::unique_ptr<data_flow::DataFlowNode>(new FillNode(	×
NEW 45	element_id, this->debug_info(), vertex, parent, this->shape_, fixed_quantization_, implementation_type_	×
NEW 46	));	×
UNCOV 47	}	×
48
NEW 49	void FillNode::validate_non_tensor_inputs(const data_flow::DataFlowGraph& graph) const {	×
NEW 50	auto* fill_value_edge = graph.in_edge_for_connector(*this, inputs_.at(1));	×
51	// Validate that the input is a scalar
NEW 52	if (fill_value_edge && fill_value_edge->base_type().type_id() != types::TypeID::Scalar) {	×
53	throw InvalidSDFGException(	×
NEW 54	"FillNode: Input memlet must be of scalar type. Found type: " + fill_value_edge->base_type().print()	×
55	);	×
56	}	×
57	}	×
58
59	} // namespace tensor
60	} // namespace math
61	} // namespace sdfg

daisytuner / docc / 26831349290

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