20282189585

Committed 16 Dec 2025 08:46PM UTC coverage: 40.083% (-0.003%) from 40.086%

Build # 20282189585

Build Type

Pull #395

github

Committed by

web-flow

Commit Message

Merge 8aa3993cd into 46cb45f72

Pull Request Pull Request #395: Various performance improvements

Run Details

13516 of 43711 branches covered (30.92%)

Branch coverage included in aggregate %.

133 of 169 new or added lines in 9 files covered. (78.7%)

9 existing lines in 4 files now uncovered.

11635 of 19037 relevant lines covered (61.12%)

87.78 hits per line

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

18.24

/src/codegen/language_extensions/c_language_extension.cpp

#include "sdfg/codegen/language_extensions/c_language_extension.h"

#include <cstddef>
#include <string>

#include "sdfg/codegen/utils.h"
#include "sdfg/data_flow/tasklet.h"
#include "sdfg/exceptions.h"
#include "sdfg/types/type.h"

namespace sdfg {
namespace codegen {

std::string CLanguageExtension::primitive_type(const types::PrimitiveType prim_type) {
    switch (prim_type) {
        case types::PrimitiveType::Void:
            return "void";
        case types::PrimitiveType::Bool:
            return "bool";
        case types::PrimitiveType::Int8:
            return "signed char";
        case types::PrimitiveType::Int16:
            return "short";
        case types::PrimitiveType::Int32:
            return "int";
        case types::PrimitiveType::Int64:
            return "long long";
        case types::PrimitiveType::Int128:
            return "__int128";
        case types::PrimitiveType::UInt8:
            return "char";
        case types::PrimitiveType::UInt16:
            return "unsigned short";
        case types::PrimitiveType::UInt32:
            return "unsigned int";
        case types::PrimitiveType::UInt64:
            return "unsigned long long";
        case types::PrimitiveType::UInt128:
            return "unsigned __int128";
        case types::PrimitiveType::Half:
            return "__fp16";
        case types::PrimitiveType::BFloat:
            return "__bf16";
        case types::PrimitiveType::Float:
            return "float";
        case types::PrimitiveType::Double:
            return "double";
        case types::PrimitiveType::X86_FP80:
            return "long double";
        case types::PrimitiveType::FP128:
            return "__float128";
        case types::PrimitiveType::PPC_FP128:
            return "__float128";
    }

    throw std::runtime_error("Unknown primitive type");
};

std::string CLanguageExtension::
    declaration(const std::string& name, const types::IType& type, bool use_initializer, bool use_alignment) {
    std::stringstream val;

    if (auto scalar_type = dynamic_cast<const types::Scalar*>(&type)) {
        val << primitive_type(scalar_type->primitive_type());
        val << " ";
        val << name;
    } else if (auto array_type = dynamic_cast<const types::Array*>(&type)) {
        auto& element_type = array_type->element_type();
        val << declaration(name + "[" + this->expression(array_type->num_elements()) + "]", element_type);
    } else if (auto pointer_type = dynamic_cast<const types::Pointer*>(&type)) {
        if (pointer_type->has_pointee_type()) {
            const types::IType& pointee = pointer_type->pointee_type();

            const bool pointee_is_function_or_array = dynamic_cast<const types::Function*>(&pointee) ||
                                                      dynamic_cast<const types::Array*>(&pointee);

            // Parenthesise *only* when it is needed to bind tighter than [] or ()
            std::string decorated = pointee_is_function_or_array ? "(*" + name + ")" : "*" + name;

            val << declaration(decorated, pointee);
        } else {
            val << "void*";
            val << " " << name;
        }
    } else if (auto ref_type = dynamic_cast<const Reference*>(&type)) {
        val << declaration("&" + name, ref_type->reference_type());
    } else if (auto structure_type = dynamic_cast<const types::Structure*>(&type)) {
        val << structure_type->name();
        val << " ";
        val << name;
    } else if (auto function_type = dynamic_cast<const types::Function*>(&type)) {
        std::stringstream params;
        for (size_t i = 0; i < function_type->num_params(); ++i) {
            params << declaration("", function_type->param_type(symbolic::integer(i)));
            if (i + 1 < function_type->num_params()) params << ", ";
        }
        if (function_type->is_var_arg()) {
            // ISO C forbids empty parameter lists before ...
            if (function_type->num_params() > 0) {
                params << ", ";
                params << "...";
            }
        }

        const std::string fun_name = name + "(" + params.str() + ")";
        val << declaration(fun_name, function_type->return_type());
    } else {
        throw std::runtime_error("Unknown declaration type");
    }

    if (use_alignment && type.alignment() > 0) {
        val << " __attribute__((aligned(" << type.alignment() << ")))";
    }

    if (use_initializer && !type.initializer().empty()) {
        val << " = " << type.initializer();
    }

    return val.str();
};

std::string CLanguageExtension::type_cast(const std::string& name, const types::IType& type) {
    std::stringstream val;

    val << "(";
    val << declaration("", type);
    val << ") ";
    val << name;

    return val.str();
};

std::string CLanguageExtension::subset(const types::IType& type, const data_flow::Subset& sub) {
    if (sub.empty()) {
        return "";
    }

    if (dynamic_cast<const types::Scalar*>(&type)) {
        return "";
    } else if (auto array_type = dynamic_cast<const types::Array*>(&type)) {
        std::string subset_str = "[" + this->expression(sub.at(0)) + "]";

        if (sub.size() > 1) {
            data_flow::Subset element_subset(sub.begin() + 1, sub.end());
            auto& element_type = array_type->element_type();
            return subset_str + subset(element_type, element_subset);
        } else {
            return subset_str;
        }
    } else if (auto pointer_type = dynamic_cast<const types::Pointer*>(&type)) {
        std::string subset_str = "[" + this->expression(sub.at(0)) + "]";

        data_flow::Subset element_subset(sub.begin() + 1, sub.end());
        auto& pointee_type = pointer_type->pointee_type();
        return subset_str + subset(pointee_type, element_subset);
    } else if (auto structure_type = dynamic_cast<const types::Structure*>(&type)) {
        auto& definition = this->function_.structure(structure_type->name());

        std::string subset_str = ".member_" + this->expression(sub.at(0));
        if (sub.size() > 1) {
            auto member = SymEngine::rcp_dynamic_cast<const SymEngine::Integer>(sub.at(0));
            auto& member_type = definition.member_type(member);
            data_flow::Subset element_subset(sub.begin() + 1, sub.end());
            return subset_str + subset(member_type, element_subset);
        } else {
            return subset_str;
        }
    }

    throw std::invalid_argument("Invalid subset type");
};

std::string CLanguageExtension::expression(const symbolic::Expression expr) {
    CSymbolicPrinter printer(this->function_, this->external_prefix_);
    return printer.apply(expr);
};

std::string CLanguageExtension::access_node(const data_flow::AccessNode& node) {
    if (dynamic_cast<const data_flow::ConstantNode*>(&node)) {
        std::string name = node.data();
        if (symbolic::is_nullptr(symbolic::symbol(name))) {
            return "NULL";
        }
        return name;
    } else {
        std::string name = node.data();
        if (this->function_.is_external(name)) {
            return "(&" + this->external_prefix_ + name + ")";
        }
        return name;
    }
};

std::string CLanguageExtension::tasklet(const data_flow::Tasklet& tasklet) {
    switch (tasklet.code()) {
        case data_flow::TaskletCode::assign:
            return tasklet.inputs().at(0);
        case data_flow::TaskletCode::fp_neg:
            return "-" + tasklet.inputs().at(0);
        case data_flow::TaskletCode::fp_add:
            return tasklet.inputs().at(0) + " + " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_sub:
            return tasklet.inputs().at(0) + " - " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_mul:
            return tasklet.inputs().at(0) + " * " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_div:
            return tasklet.inputs().at(0) + " / " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_rem:
            return "remainder(" + tasklet.inputs().at(0) + ", " + tasklet.inputs().at(1) + ")";
        case data_flow::TaskletCode::fp_fma:
            return tasklet.inputs().at(0) + " * " + tasklet.inputs().at(1) + " + " + tasklet.inputs().at(2);
        case data_flow::TaskletCode::fp_oeq:
            return tasklet.inputs().at(0) + " == " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_one:
            return tasklet.inputs().at(0) + " != " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_ogt:
            return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_oge:
            return tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_olt:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_ole:
            return tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_ord:
            return "isnan(" + tasklet.inputs().at(0) + ") && isnan(" + tasklet.inputs().at(1) + ")";
        case data_flow::TaskletCode::fp_ueq:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")" + " || " +
                   tasklet.inputs().at(0) + " == " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_une:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")" + " || " +
                   tasklet.inputs().at(0) + " != " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_ugt:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")" + " || " +
                   tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_uge:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")" + " || " +
                   tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_ult:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")" + " || " +
                   tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_ule:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")" + " || " +
                   tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::fp_uno:
            return "isnan(" + tasklet.inputs().at(0) + ") || isnan(" + tasklet.inputs().at(1) + ")";
        case data_flow::TaskletCode::int_add:
            return tasklet.inputs().at(0) + " + " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_sub:
            return tasklet.inputs().at(0) + " - " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_mul:
            return tasklet.inputs().at(0) + " * " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_sdiv:
            return tasklet.inputs().at(0) + " / " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_srem:
            return tasklet.inputs().at(0) + " % " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_udiv:
            return tasklet.inputs().at(0) + " / " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_urem:
            return tasklet.inputs().at(0) + " % " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_and:
            return tasklet.inputs().at(0) + " & " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_or:
            return tasklet.inputs().at(0) + " | " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_xor:
            return tasklet.inputs().at(0) + " ^ " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_shl:
            return tasklet.inputs().at(0) + " << " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_lshr:
            return tasklet.inputs().at(0) + " >> " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_ashr:
            return tasklet.inputs().at(0) + " >> " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_smin:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +
                   tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_smax:
            return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +
                   tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_scmp:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? -1 : (" + tasklet.inputs().at(0) +
                   " > " + tasklet.inputs().at(1) + " ? 1 : 0)";
        case data_flow::TaskletCode::int_umin:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +
                   tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_umax:
            return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +
                   tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_ucmp:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? -1 : (" + tasklet.inputs().at(0) +
                   " > " + tasklet.inputs().at(1) + " ? 1 : 0)";
        case data_flow::TaskletCode::int_abs:
            return "(" + tasklet.inputs().at(0) + " < 0 ? -" + tasklet.inputs().at(0) + " : " + tasklet.inputs().at(0) +
                   ")";
        case data_flow::TaskletCode::int_eq:
            return tasklet.inputs().at(0) + " == " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_ne:
            return tasklet.inputs().at(0) + " != " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_sgt:
            return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_sge:
            return tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_slt:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_sle:
            return tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_ugt:
            return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_uge:
            return tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_ult:
            return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);
        case data_flow::TaskletCode::int_ule:
            return tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);
    };
    throw std::invalid_argument("Invalid tasklet code");
};

std::string CLanguageExtension::zero(const types::PrimitiveType prim_type) {
    switch (prim_type) {
        case types::Void:
            throw InvalidSDFGException("No zero for void type possible");
        case types::Bool:
            return "false";
        case types::Int8:
            return "0";
        case types::Int16:
            return "0";
        case types::Int32:
            return "0";
        case types::Int64:
            return "0ll";
        case types::Int128:
            return "0";
        case types::UInt8:
            return "0u";
        case types::UInt16:
            return "0u";
        case types::UInt32:
            return "0u";
        case types::UInt64:
            return "0ull";
        case types::UInt128:
            return "0u";
        case types::Half:
            throw InvalidSDFGException("Currently unsupported");
        case types::BFloat:
            throw InvalidSDFGException("Currently unsupported");
        case types::Float:
            return "0.0f";
        case types::Double:
            return "0.0";
        case types::X86_FP80:
            return "0.0l";
        case types::FP128:
            throw InvalidSDFGException("Currently unsupported");
        case types::PPC_FP128:
            throw InvalidSDFGException("Currently unsupported");
    }
}

void CSymbolicPrinter::bvisit(const SymEngine::Infty& x) {
    if (x.is_negative_infinity())
        str_ = "-INFINITY";
    else if (x.is_positive_infinity())
        str_ = "INFINITY";
};

void CSymbolicPrinter::bvisit(const SymEngine::BooleanAtom& x) { str_ = x.get_val() ? "true" : "false"; };

void CSymbolicPrinter::bvisit(const SymEngine::Symbol& x) {
    if (symbolic::is_nullptr(symbolic::symbol(x.get_name()))) {
        str_ = "((uintptr_t) NULL)";
        return;
    }

    std::string name = x.get_name();

    if (this->function_.is_external(name)) {
        name = "((uintptr_t) (&" + this->external_prefix_ + name + "))";
    } else if (this->function_.exists(name) && this->function_.type(name).type_id() == types::TypeID::Pointer) {
        name = "((uintptr_t) " + name + ")";
    }

    str_ = name;
};

void CSymbolicPrinter::bvisit(const SymEngine::And& x) {
    std::ostringstream s;
    auto container = x.get_container();
    s << apply(*container.begin());
    for (auto it = ++(container.begin()); it != container.end(); ++it) {
        s << " && " << apply(*it);
    }
    str_ = parenthesize(s.str());
};

void CSymbolicPrinter::bvisit(const SymEngine::Or& x) {
    std::ostringstream s;
    auto container = x.get_container();
    s << apply(*container.begin());
    for (auto it = ++(container.begin()); it != container.end(); ++it) {
        s << " || " << apply(*it);
    }
    str_ = parenthesize(s.str());
};

void CSymbolicPrinter::bvisit(const SymEngine::Not& x) {
    str_ = "!" + apply(x.get_arg());
    str_ = parenthesize(str_);
};

void CSymbolicPrinter::bvisit(const SymEngine::Equality& x) {
    str_ = apply(x.get_args()[0]) + " == " + apply(x.get_args()[1]);
    str_ = parenthesize(str_);
};

void CSymbolicPrinter::bvisit(const SymEngine::Unequality& x) {
    str_ = apply(x.get_args()[0]) + " != " + apply(x.get_args()[1]);
    str_ = parenthesize(str_);
};

void CSymbolicPrinter::bvisit(const SymEngine::Min& x) {
    std::ostringstream s;
    auto container = x.get_args();
    if (container.size() == 1) {
        s << apply(*container.begin());
    } else {
        if (this->use_rtl_functions_) {
            s << "__daisy_min(";
        } else {
            s << "min(";
        }
        s << apply(*container.begin());

        // Recursively apply __daisy_min to the arguments
        SymEngine::vec_basic subargs;
        for (auto it = ++(container.begin()); it != container.end(); ++it) {
            subargs.push_back(*it);
        }
        auto submin = SymEngine::min(subargs);
        s << ", " << apply(submin);

        s << ")";
    }

    str_ = s.str();
};

void CSymbolicPrinter::bvisit(const SymEngine::Max& x) {
    std::ostringstream s;
    auto container = x.get_args();
    if (container.size() == 1) {
        s << apply(*container.begin());
    } else {
        if (this->use_rtl_functions_) {
            s << "__daisy_max(";
        } else {
            s << "max(";
        }
        s << apply(*container.begin());

        // Recursively apply __daisy_max to the arguments
        SymEngine::vec_basic subargs;
        for (auto it = ++(container.begin()); it != container.end(); ++it) {
            subargs.push_back(*it);
        }
        auto submax = SymEngine::max(subargs);
        s << ", " << apply(submax);

        s << ")";
    }

    str_ = s.str();
};

void CSymbolicPrinter::bvisit(const SymEngine::FunctionSymbol& x) {
    if (x.get_name() == "idiv") {
        str_ = "((" + apply(x.get_args()[0]) + ") / (" + apply(x.get_args()[1]) + "))";
    } else if (x.get_name() == "iabs") {
        str_ = "((" + apply(x.get_args()[0]) + ") < 0 ? -(" + apply(x.get_args()[0]) + ") : (" +
               apply(x.get_args()[0]) + "))";
    } else if (x.get_name() == "imod") {
        str_ = "((" + apply(x.get_args()[0]) + ") % (" + apply(x.get_args()[1]) + "))";
    } else if (x.get_name() == "sizeof") {
        auto& so = dynamic_cast<const symbolic::SizeOfTypeFunction&>(x);
        auto& type = so.get_type();
        CLanguageExtension lang(this->function_, this->external_prefix_);
        str_ = "sizeof(" + lang.declaration("", type) + ")";
    } else if (x.get_name() == "malloc_usable_size") {
        str_ = "malloc_usable_size(" +
               SymEngine::rcp_static_cast<const SymEngine::Symbol>(x.get_args()[0])->get_name() + ")";
    } else {
        throw std::runtime_error("Unsupported function symbol: " + x.get_name());
    }
};

void CSymbolicPrinter::_print_pow(
    std::ostringstream& o,
    const SymEngine::RCP<const SymEngine::Basic>& a,
    const SymEngine::RCP<const SymEngine::Basic>& b
) {
    if (SymEngine::eq(*a, *SymEngine::E)) {
        o << "exp(" << apply(b) << ")";
    } else if (SymEngine::eq(*b, *SymEngine::rational(1, 2))) {
        o << "sqrt(" << apply(a) << ")";
    } else if (SymEngine::eq(*b, *SymEngine::rational(1, 3))) {
        o << "cbrt(" << apply(a) << ")";
    } else if (SymEngine::eq(*b, *SymEngine::integer(2))) {
        o << "((" + apply(a) + ") * (" + apply(a) + "))";
    } else {
        o << "__daisy_sym_pow(" << apply(a) << ", " << apply(b) << ")";
    }
};

} // namespace codegen
} // namespace sdfg

1	#include "sdfg/codegen/language_extensions/c_language_extension.h"
2
3	#include <cstddef>
4	#include <string>
5
6	#include "sdfg/codegen/utils.h"
7	#include "sdfg/data_flow/tasklet.h"
8	#include "sdfg/exceptions.h"
9	#include "sdfg/types/type.h"
10
11	namespace sdfg {
12	namespace codegen {
13
14	std::string CLanguageExtension::primitive_type(const types::PrimitiveType prim_type) {	35✔
15	switch (prim_type) {	35!
16	case types::PrimitiveType::Void:
17	return "void";	2!
18	case types::PrimitiveType::Bool:
19	return "bool";	1!
20	case types::PrimitiveType::Int8:
21	return "signed char";	1!
22	case types::PrimitiveType::Int16:
23	return "short";	1!
24	case types::PrimitiveType::Int32:
25	return "int";	17!
26	case types::PrimitiveType::Int64:
27	return "long long";	4!
28	case types::PrimitiveType::Int128:
29	return "__int128";	×
30	case types::PrimitiveType::UInt8:
31	return "char";	3!
32	case types::PrimitiveType::UInt16:
33	return "unsigned short";	1!
34	case types::PrimitiveType::UInt32:
35	return "unsigned int";	1!
36	case types::PrimitiveType::UInt64:
37	return "unsigned long long";	1!
38	case types::PrimitiveType::UInt128:
39	return "unsigned __int128";	×
40	case types::PrimitiveType::Half:
41	return "__fp16";	×
42	case types::PrimitiveType::BFloat:
43	return "__bf16";	×
44	case types::PrimitiveType::Float:
45	return "float";	2!
46	case types::PrimitiveType::Double:
47	return "double";	1!
48	case types::PrimitiveType::X86_FP80:
49	return "long double";	×
50	case types::PrimitiveType::FP128:
51	return "__float128";	×
52	case types::PrimitiveType::PPC_FP128:
53	return "__float128";	×
54	}
55
56	throw std::runtime_error("Unknown primitive type");	×
57	};	35✔
58
59	std::string CLanguageExtension::
60	declaration(const std::string& name, const types::IType& type, bool use_initializer, bool use_alignment) {	43✔
61	std::stringstream val;	43✔
62
63	if (auto scalar_type = dynamic_cast<const types::Scalar*>(&type)) {	43!
64	val << primitive_type(scalar_type->primitive_type());	23!
65	val << " ";	23!
66	val << name;	23!
67	} else if (auto array_type = dynamic_cast<const types::Array*>(&type)) {	43!
68	auto& element_type = array_type->element_type();	3!
69	val << declaration(name + "[" + this->expression(array_type->num_elements()) + "]", element_type);	3!
70	} else if (auto pointer_type = dynamic_cast<const types::Pointer*>(&type)) {	20!
71	if (pointer_type->has_pointee_type()) {	13!
72	const types::IType& pointee = pointer_type->pointee_type();	12!
73
74	const bool pointee_is_function_or_array = dynamic_cast<const types::Function*>(&pointee) \|\|	24!
75	dynamic_cast<const types::Array*>(&pointee);	12!
76
77	// Parenthesise only when it is needed to bind tighter than [] or ()
78	std::string decorated = pointee_is_function_or_array ? "(" + name + ")" : "" + name;	12!
79
80	val << declaration(decorated, pointee);	12!
81	} else {	12✔
82	val << "void*";	1!
83	val << " " << name;	1!
84	}
85	} else if (auto ref_type = dynamic_cast<const Reference*>(&type)) {	17!
86	val << declaration("&" + name, ref_type->reference_type());	×
87	} else if (auto structure_type = dynamic_cast<const types::Structure*>(&type)) {	4!
88	val << structure_type->name();	4!
89	val << " ";	4!
90	val << name;	4!
91	} else if (auto function_type = dynamic_cast<const types::Function*>(&type)) {	4!
92	std::stringstream params;	×
93	for (size_t i = 0; i < function_type->num_params(); ++i) {	×
94	params << declaration("", function_type->param_type(symbolic::integer(i)));	×
95	if (i + 1 < function_type->num_params()) params << ", ";	×
96	}	×
97	if (function_type->is_var_arg()) {	×
98	// ISO C forbids empty parameter lists before ...
99	if (function_type->num_params() > 0) {	×
100	params << ", ";	×
101	params << "...";	×
102	}	×
103	}	×
104
105	const std::string fun_name = name + "(" + params.str() + ")";	×
106	val << declaration(fun_name, function_type->return_type());	×
107	} else {	×
108	throw std::runtime_error("Unknown declaration type");	×
109	}
110
111	if (use_alignment && type.alignment() > 0) {	43!
112	val << " __attribute__((aligned(" << type.alignment() << ")))";	×
113	}	×
114
115	if (use_initializer && !type.initializer().empty()) {	43!
116	val << " = " << type.initializer();	×
117	}	×
118
119	return val.str();	43!
120	};	43✔
121
122	std::string CLanguageExtension::type_cast(const std::string& name, const types::IType& type) {	5✔
123	std::stringstream val;	5✔
124
125	val << "(";	5!
126	val << declaration("", type);	5!
127	val << ") ";	5!
128	val << name;	5!
129
130	return val.str();	5!
131	};	5✔
132
133	std::string CLanguageExtension::subset(const types::IType& type, const data_flow::Subset& sub) {	15✔
134	if (sub.empty()) {	15✔
135	return "";	12!
136	}
137
138	if (dynamic_cast<const types::Scalar*>(&type)) {	3!
139	return "";	×
140	} else if (auto array_type = dynamic_cast<const types::Array*>(&type)) {	3!
141	std::string subset_str = "[" + this->expression(sub.at(0)) + "]";	1!
142
143	if (sub.size() > 1) {	1!
144	data_flow::Subset element_subset(sub.begin() + 1, sub.end());	×
145	auto& element_type = array_type->element_type();	×
146	return subset_str + subset(element_type, element_subset);	×
147	} else {	×
148	return subset_str;	1✔
149	}
150	} else if (auto pointer_type = dynamic_cast<const types::Pointer*>(&type)) {	3!
151	std::string subset_str = "[" + this->expression(sub.at(0)) + "]";	1!
152
153	data_flow::Subset element_subset(sub.begin() + 1, sub.end());	1!
154	auto& pointee_type = pointer_type->pointee_type();	1!
155	return subset_str + subset(pointee_type, element_subset);	1!
156	} else if (auto structure_type = dynamic_cast<const types::Structure*>(&type)) {	2!
157	auto& definition = this->function_.structure(structure_type->name());	1✔
158
159	std::string subset_str = ".member_" + this->expression(sub.at(0));	1!
160	if (sub.size() > 1) {	1!
161	auto member = SymEngine::rcp_dynamic_cast<const SymEngine::Integer>(sub.at(0));	×
162	auto& member_type = definition.member_type(member);	×
163	data_flow::Subset element_subset(sub.begin() + 1, sub.end());	×
164	return subset_str + subset(member_type, element_subset);	×
165	} else {	×
166	return subset_str;	1✔
167	}
168	}	1✔
169
170	throw std::invalid_argument("Invalid subset type");	×
171	};	15✔
172
173	std::string CLanguageExtension::expression(const symbolic::Expression expr) {	40✔
174	CSymbolicPrinter printer(this->function_, this->external_prefix_);	40✔
175	return printer.apply(expr);	40!
176	};	40✔
177
178	std::string CLanguageExtension::access_node(const data_flow::AccessNode& node) {	21✔
179	if (dynamic_cast<const data_flow::ConstantNode*>(&node)) {	21!
180	std::string name = node.data();	3✔
181	if (symbolic::is_nullptr(symbolic::symbol(name))) {	3!
182	return "NULL";	2!
183	}
184	return name;	1✔
185	} else {	3✔
186	std::string name = node.data();	18✔
187	if (this->function_.is_external(name)) {	18!
188	return "(&" + this->external_prefix_ + name + ")";	×
189	}
190	return name;	18✔
191	}	18✔
192	};	21✔
193
194	std::string CLanguageExtension::tasklet(const data_flow::Tasklet& tasklet) {	3✔
195	switch (tasklet.code()) {	3!
196	case data_flow::TaskletCode::assign:
197	return tasklet.inputs().at(0);	2✔
198	case data_flow::TaskletCode::fp_neg:
199	return "-" + tasklet.inputs().at(0);	×
200	case data_flow::TaskletCode::fp_add:
201	return tasklet.inputs().at(0) + " + " + tasklet.inputs().at(1);	×
202	case data_flow::TaskletCode::fp_sub:
203	return tasklet.inputs().at(0) + " - " + tasklet.inputs().at(1);	×
204	case data_flow::TaskletCode::fp_mul:
205	return tasklet.inputs().at(0) + " * " + tasklet.inputs().at(1);	×
206	case data_flow::TaskletCode::fp_div:
207	return tasklet.inputs().at(0) + " / " + tasklet.inputs().at(1);	×
208	case data_flow::TaskletCode::fp_rem:
209	return "remainder(" + tasklet.inputs().at(0) + ", " + tasklet.inputs().at(1) + ")";	×
210	case data_flow::TaskletCode::fp_fma:
211	return tasklet.inputs().at(0) + " * " + tasklet.inputs().at(1) + " + " + tasklet.inputs().at(2);	×
212	case data_flow::TaskletCode::fp_oeq:
213	return tasklet.inputs().at(0) + " == " + tasklet.inputs().at(1);	×
214	case data_flow::TaskletCode::fp_one:
215	return tasklet.inputs().at(0) + " != " + tasklet.inputs().at(1);	×
216	case data_flow::TaskletCode::fp_ogt:
217	return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);	×
218	case data_flow::TaskletCode::fp_oge:
219	return tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);	×
220	case data_flow::TaskletCode::fp_olt:
221	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);	×
222	case data_flow::TaskletCode::fp_ole:
223	return tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);	×
224	case data_flow::TaskletCode::fp_ord:
225	return "isnan(" + tasklet.inputs().at(0) + ") && isnan(" + tasklet.inputs().at(1) + ")";	×
226	case data_flow::TaskletCode::fp_ueq:
227	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")" + " \|\| " +	×
228	tasklet.inputs().at(0) + " == " + tasklet.inputs().at(1);	×
229	case data_flow::TaskletCode::fp_une:
230	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")" + " \|\| " +	×
231	tasklet.inputs().at(0) + " != " + tasklet.inputs().at(1);	×
232	case data_flow::TaskletCode::fp_ugt:
233	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")" + " \|\| " +	×
234	tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);	×
235	case data_flow::TaskletCode::fp_uge:
236	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")" + " \|\| " +	×
237	tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);	×
238	case data_flow::TaskletCode::fp_ult:
239	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")" + " \|\| " +	×
240	tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);	×
241	case data_flow::TaskletCode::fp_ule:
242	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")" + " \|\| " +	×
243	tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);	×
244	case data_flow::TaskletCode::fp_uno:
245	return "isnan(" + tasklet.inputs().at(0) + ") \|\| isnan(" + tasklet.inputs().at(1) + ")";	×
246	case data_flow::TaskletCode::int_add:
247	return tasklet.inputs().at(0) + " + " + tasklet.inputs().at(1);	1!
248	case data_flow::TaskletCode::int_sub:
249	return tasklet.inputs().at(0) + " - " + tasklet.inputs().at(1);	×
250	case data_flow::TaskletCode::int_mul:
251	return tasklet.inputs().at(0) + " * " + tasklet.inputs().at(1);	×
252	case data_flow::TaskletCode::int_sdiv:
253	return tasklet.inputs().at(0) + " / " + tasklet.inputs().at(1);	×
254	case data_flow::TaskletCode::int_srem:
255	return tasklet.inputs().at(0) + " % " + tasklet.inputs().at(1);	×
256	case data_flow::TaskletCode::int_udiv:
257	return tasklet.inputs().at(0) + " / " + tasklet.inputs().at(1);	×
258	case data_flow::TaskletCode::int_urem:
259	return tasklet.inputs().at(0) + " % " + tasklet.inputs().at(1);	×
260	case data_flow::TaskletCode::int_and:
261	return tasklet.inputs().at(0) + " & " + tasklet.inputs().at(1);	×
262	case data_flow::TaskletCode::int_or:
263	return tasklet.inputs().at(0) + " \| " + tasklet.inputs().at(1);	×
264	case data_flow::TaskletCode::int_xor:
265	return tasklet.inputs().at(0) + " ^ " + tasklet.inputs().at(1);	×
266	case data_flow::TaskletCode::int_shl:
267	return tasklet.inputs().at(0) + " << " + tasklet.inputs().at(1);	×
268	case data_flow::TaskletCode::int_lshr:
269	return tasklet.inputs().at(0) + " >> " + tasklet.inputs().at(1);	×
270	case data_flow::TaskletCode::int_ashr:
271	return tasklet.inputs().at(0) + " >> " + tasklet.inputs().at(1);	×
272	case data_flow::TaskletCode::int_smin:
273	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +	×
274	tasklet.inputs().at(1);	×
275	case data_flow::TaskletCode::int_smax:
276	return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +	×
277	tasklet.inputs().at(1);	×
278	case data_flow::TaskletCode::int_scmp:
279	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? -1 : (" + tasklet.inputs().at(0) +	×
280	" > " + tasklet.inputs().at(1) + " ? 1 : 0)";	×
281	case data_flow::TaskletCode::int_umin:
282	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +	×
283	tasklet.inputs().at(1);	×
284	case data_flow::TaskletCode::int_umax:
285	return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1) + " ? " + tasklet.inputs().at(0) + " : " +	×
286	tasklet.inputs().at(1);	×
287	case data_flow::TaskletCode::int_ucmp:
288	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1) + " ? -1 : (" + tasklet.inputs().at(0) +	×
289	" > " + tasklet.inputs().at(1) + " ? 1 : 0)";	×
290	case data_flow::TaskletCode::int_abs:
291	return "(" + tasklet.inputs().at(0) + " < 0 ? -" + tasklet.inputs().at(0) + " : " + tasklet.inputs().at(0) +	×
292	")";
293	case data_flow::TaskletCode::int_eq:
294	return tasklet.inputs().at(0) + " == " + tasklet.inputs().at(1);	×
295	case data_flow::TaskletCode::int_ne:
296	return tasklet.inputs().at(0) + " != " + tasklet.inputs().at(1);	×
297	case data_flow::TaskletCode::int_sgt:
298	return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);	×
299	case data_flow::TaskletCode::int_sge:
300	return tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);	×
301	case data_flow::TaskletCode::int_slt:
302	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);	×
303	case data_flow::TaskletCode::int_sle:
304	return tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);	×
305	case data_flow::TaskletCode::int_ugt:
306	return tasklet.inputs().at(0) + " > " + tasklet.inputs().at(1);	×
307	case data_flow::TaskletCode::int_uge:
308	return tasklet.inputs().at(0) + " >= " + tasklet.inputs().at(1);	×
309	case data_flow::TaskletCode::int_ult:
310	return tasklet.inputs().at(0) + " < " + tasklet.inputs().at(1);	×
311	case data_flow::TaskletCode::int_ule:
312	return tasklet.inputs().at(0) + " <= " + tasklet.inputs().at(1);	×
313	};
314	throw std::invalid_argument("Invalid tasklet code");	×
315	};	3✔
316
317	std::string CLanguageExtension::zero(const types::PrimitiveType prim_type) {	×
318	switch (prim_type) {	×
319	case types::Void:
320	throw InvalidSDFGException("No zero for void type possible");	×
321	case types::Bool:
322	return "false";	×
323	case types::Int8:
324	return "0";	×
325	case types::Int16:
326	return "0";	×
327	case types::Int32:
328	return "0";	×
329	case types::Int64:
330	return "0ll";	×
331	case types::Int128:
332	return "0";	×
333	case types::UInt8:
334	return "0u";	×
335	case types::UInt16:
336	return "0u";	×
337	case types::UInt32:
338	return "0u";	×
339	case types::UInt64:
340	return "0ull";	×
341	case types::UInt128:
342	return "0u";	×
343	case types::Half:
344	throw InvalidSDFGException("Currently unsupported");	×
345	case types::BFloat:
346	throw InvalidSDFGException("Currently unsupported");	×
347	case types::Float:
348	return "0.0f";	×
349	case types::Double:
350	return "0.0";	×
351	case types::X86_FP80:
352	return "0.0l";	×
353	case types::FP128:
354	throw InvalidSDFGException("Currently unsupported");	×
355	case types::PPC_FP128:
356	throw InvalidSDFGException("Currently unsupported");	×
357	}	×
358	}	×
359
360	void CSymbolicPrinter::bvisit(const SymEngine::Infty& x) {	×
361	if (x.is_negative_infinity())	×
362	str_ = "-INFINITY";	×
363	else if (x.is_positive_infinity())	×
364	str_ = "INFINITY";	×
365	};	×
366
367	void CSymbolicPrinter::bvisit(const SymEngine::BooleanAtom& x) { str_ = x.get_val() ? "true" : "false"; };	2✔
368
369	void CSymbolicPrinter::bvisit(const SymEngine::Symbol& x) {	1,071✔
370	if (symbolic::is_nullptr(symbolic::symbol(x.get_name()))) {	1,071!
371	str_ = "((uintptr_t) NULL)";	×
372	return;	×
373	}
374
375	std::string name = x.get_name();	1,071✔
376
377	if (this->function_.is_external(name)) {	1,071!
378	name = "((uintptr_t) (&" + this->external_prefix_ + name + "))";	1!
379	} else if (this->function_.exists(name) && this->function_.type(name).type_id() == types::TypeID::Pointer) {	1,071!
380	name = "((uintptr_t) " + name + ")";	×
381	}	×
382
383	str_ = name;	1,071!
384	};	1,071✔
385
386	void CSymbolicPrinter::bvisit(const SymEngine::And& x) {	×
387	std::ostringstream s;	×
388	auto container = x.get_container();	×
389	s << apply(*container.begin());	×
390	for (auto it = ++(container.begin()); it != container.end(); ++it) {	×
391	s << " && " << apply(*it);	×
392	}	×
393	str_ = parenthesize(s.str());	×
394	};	×
395
396	void CSymbolicPrinter::bvisit(const SymEngine::Or& x) {	×
397	std::ostringstream s;	×
398	auto container = x.get_container();	×
399	s << apply(*container.begin());	×
400	for (auto it = ++(container.begin()); it != container.end(); ++it) {	×
401	s << " \|\| " << apply(*it);	×
402	}	×
403	str_ = parenthesize(s.str());	×
404	};	×
405
406	void CSymbolicPrinter::bvisit(const SymEngine::Not& x) {	×
407	str_ = "!" + apply(x.get_arg());	×
408	str_ = parenthesize(str_);	×
409	};	×
410
411	void CSymbolicPrinter::bvisit(const SymEngine::Equality& x) {	1✔
412	str_ = apply(x.get_args()[0]) + " == " + apply(x.get_args()[1]);	1!
413	str_ = parenthesize(str_);	1✔
414	};	1✔
415
416	void CSymbolicPrinter::bvisit(const SymEngine::Unequality& x) {	1✔
417	str_ = apply(x.get_args()[0]) + " != " + apply(x.get_args()[1]);	1!
418	str_ = parenthesize(str_);	1✔
419	};	1✔
420
421	void CSymbolicPrinter::bvisit(const SymEngine::Min& x) {	×
422	std::ostringstream s;	×
423	auto container = x.get_args();	×
424	if (container.size() == 1) {	×
425	s << apply(*container.begin());	×
426	} else {	×
NEW 427	if (this->use_rtl_functions_) {	×
NEW 428	s << "__daisy_min(";	×
NEW 429	} else {	×
NEW 430	s << "min(";	×
431	}
UNCOV 432	s << apply(*container.begin());	×
433
434	// Recursively apply __daisy_min to the arguments
435	SymEngine::vec_basic subargs;	×
436	for (auto it = ++(container.begin()); it != container.end(); ++it) {	×
437	subargs.push_back(*it);	×
438	}	×
439	auto submin = SymEngine::min(subargs);	×
440	s << ", " << apply(submin);	×
441
442	s << ")";	×
443	}	×
444
445	str_ = s.str();	×
446	};	×
447
448	void CSymbolicPrinter::bvisit(const SymEngine::Max& x) {	×
449	std::ostringstream s;	×
450	auto container = x.get_args();	×
451	if (container.size() == 1) {	×
452	s << apply(*container.begin());	×
453	} else {	×
NEW 454	if (this->use_rtl_functions_) {	×
NEW 455	s << "__daisy_max(";	×
NEW 456	} else {	×
NEW 457	s << "max(";	×
458	}
UNCOV 459	s << apply(*container.begin());	×
460
461	// Recursively apply __daisy_max to the arguments
462	SymEngine::vec_basic subargs;	×
463	for (auto it = ++(container.begin()); it != container.end(); ++it) {	×
464	subargs.push_back(*it);	×
465	}	×
466	auto submax = SymEngine::max(subargs);	×
467	s << ", " << apply(submax);	×
468
469	s << ")";	×
470	}	×
471
472	str_ = s.str();	×
473	};	×
474
475	void CSymbolicPrinter::bvisit(const SymEngine::FunctionSymbol& x) {	1✔
476	if (x.get_name() == "idiv") {	1!
477	str_ = "((" + apply(x.get_args()[0]) + ") / (" + apply(x.get_args()[1]) + "))";	×
478	} else if (x.get_name() == "iabs") {	1!
479	str_ = "((" + apply(x.get_args()[0]) + ") < 0 ? -(" + apply(x.get_args()[0]) + ") : (" +	×
480	apply(x.get_args()[0]) + "))";	×
481	} else if (x.get_name() == "imod") {	1!
482	str_ = "((" + apply(x.get_args()[0]) + ") % (" + apply(x.get_args()[1]) + "))";	×
483	} else if (x.get_name() == "sizeof") {	1!
484	auto& so = dynamic_cast<const symbolic::SizeOfTypeFunction&>(x);	1!
485	auto& type = so.get_type();	1✔
486	CLanguageExtension lang(this->function_, this->external_prefix_);	1✔
487	str_ = "sizeof(" + lang.declaration("", type) + ")";	1!
488	} else if (x.get_name() == "malloc_usable_size") {	1!
NEW 489	str_ = "malloc_usable_size(" +	×
NEW 490	SymEngine::rcp_static_cast<const SymEngine::Symbol>(x.get_args()[0])->get_name() + ")";	×
491	} else {	×
492	throw std::runtime_error("Unsupported function symbol: " + x.get_name());	×
493	}
494	};	1✔
495
496	void CSymbolicPrinter::_print_pow(	2✔
497	std::ostringstream& o,
498	const SymEngine::RCP<const SymEngine::Basic>& a,
499	const SymEngine::RCP<const SymEngine::Basic>& b
500	) {
501	if (SymEngine::eq(a, SymEngine::E)) {	2!
502	o << "exp(" << apply(b) << ")";	×
503	} else if (SymEngine::eq(b, SymEngine::rational(1, 2))) {	2!
504	o << "sqrt(" << apply(a) << ")";	×
505	} else if (SymEngine::eq(b, SymEngine::rational(1, 3))) {	2!
506	o << "cbrt(" << apply(a) << ")";	×
507	} else if (SymEngine::eq(b, SymEngine::integer(2))) {	2!
508	o << "((" + apply(a) + ") * (" + apply(a) + "))";	2!
509	} else {	2✔
510	o << "__daisy_sym_pow(" << apply(a) << ", " << apply(b) << ")";	×
511	}
512	};	2✔
513
514	} // namespace codegen
515	} // namespace sdfg

daisytuner / sdfglib / 20282189585

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