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

0.0

/opt/src/targets/tenstorrent/library_node_mapping.cpp

#include "sdfg/targets/tenstorrent/library_node_mapping.h"

#include "sdfg/data_flow/library_nodes/math/blas/dot_node.h"
#include "sdfg/data_flow/library_nodes/math/blas/gemm_node.h"

#include "sdfg/targets/tenstorrent/plugin.h"

namespace sdfg::tenstorrent {

std::optional<data_flow::ImplementationType> try_map_library_node_implementation(const data_flow::LibraryNodeCode& code
) {
    if (code == math::blas::LibraryNodeType_GEMM.value()) {
        return ImplementationType_Tenstorrent_WithTransfers;
    } else if (code == math::blas::LibraryNodeType_DOT.value()) {
        return ImplementationType_Tenstorrent_WithTransfers;
    } else {
        return std::nullopt;
    }
}

std::optional<data_flow::ImplementationType>
try_map_library_node_implementation(const data_flow::LibraryNodeCode& code, types::PrimitiveType data_type) {
    if (data_type == types::PrimitiveType::Float) {
        if (code == math::blas::LibraryNodeType_GEMM.value()) {
            return ImplementationType_Tenstorrent_WithTransfers;
        } else if (code == math::blas::LibraryNodeType_DOT.value()) {
            return ImplementationType_Tenstorrent_WithTransfers;
        } else {
            return std::nullopt;
        }
    } else {
        return std::nullopt;
    }
}

std::optional<data_flow::ImplementationType> try_map_blas_node_implementation(const math::blas::BLASNode& node) {
    auto data_type = node.scalar_primitive();
    auto& code = node.code();

    return try_map_library_node_implementation(code, data_type);
}

} // namespace sdfg::tenstorrent

1	#include "sdfg/targets/tenstorrent/library_node_mapping.h"
2
3	#include "sdfg/data_flow/library_nodes/math/blas/dot_node.h"
4	#include "sdfg/data_flow/library_nodes/math/blas/gemm_node.h"
5
6	#include "sdfg/targets/tenstorrent/plugin.h"
7
8	namespace sdfg::tenstorrent {
9
10	std::optional<data_flow::ImplementationType> try_map_library_node_implementation(const data_flow::LibraryNodeCode& code
NEW 11	) {	×
NEW 12	if (code == math::blas::LibraryNodeType_GEMM.value()) {	×
NEW 13	return ImplementationType_Tenstorrent_WithTransfers;	×
NEW 14	} else if (code == math::blas::LibraryNodeType_DOT.value()) {	×
NEW 15	return ImplementationType_Tenstorrent_WithTransfers;	×
NEW 16	} else {	×
NEW 17	return std::nullopt;	×
NEW 18	}	×
NEW 19	}	×
20
21	std::optional<data_flow::ImplementationType>
NEW 22	try_map_library_node_implementation(const data_flow::LibraryNodeCode& code, types::PrimitiveType data_type) {	×
NEW 23	if (data_type == types::PrimitiveType::Float) {	×
NEW 24	if (code == math::blas::LibraryNodeType_GEMM.value()) {	×
NEW 25	return ImplementationType_Tenstorrent_WithTransfers;	×
NEW 26	} else if (code == math::blas::LibraryNodeType_DOT.value()) {	×
NEW 27	return ImplementationType_Tenstorrent_WithTransfers;	×
NEW 28	} else {	×
NEW 29	return std::nullopt;	×
NEW 30	}	×
NEW 31	} else {	×
NEW 32	return std::nullopt;	×
NEW 33	}	×
NEW 34	}	×
35
NEW 36	std::optional<data_flow::ImplementationType> try_map_blas_node_implementation(const math::blas::BLASNode& node) {	×
NEW 37	auto data_type = node.scalar_primitive();	×
NEW 38	auto& code = node.code();	×
39
NEW 40	return try_map_library_node_implementation(code, data_type);	×
NEW 41	}	×
42
43	} // namespace sdfg::tenstorrent

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