TensorConversion.h File

Conversion vocabulary, policies, and tracing for Tensor. More...

Included Headers

#include "pipeline/TensorCore.h" #include <cstdint> #include <string> #include <vector>

Namespaces Index

namespace	simaai
namespace	neat

Classes Index

struct	ConversionCost
struct	ConversionTrace
struct	ConversionTraceCollector

Description

Conversion vocabulary, policies, and tracing for Tensor.

File Listing

The file content with the documentation metadata removed is:

1

6#pragma once

7

8#include "pipeline/TensorCore.h"

9

10#include <cstdint>

11#include <string>

12#include <vector>

13

14namespace simaai::neat {

15

16enum class ConversionKind {

17 Reinterpret = 0,

18 View,

19 Pack,

20 Convert,

21 Transfer,

22};

23

24enum class ConversionPolicy {

25 Strict = 0,

26 AllowWithTrace,

27 AllowSilent,

28};

29

30struct ConversionCost {

31 std::uint64_t bytes_copied = 0;

32 int compute_class = 0; // 0=low, 1=med, 2=high

33};

34

35struct ConversionTrace {

36 std::string stage;

37 ConversionKind kind = ConversionKind::Reinterpret;

38 std::string src_desc;

39 std::string dst_desc;

40 std::uint64_t bytes_copied = 0;

41 std::uint64_t elapsed_us = 0;

42 ConversionPolicy policy = ConversionPolicy::Strict;

43};

44

45struct ConversionTraceCollector {

46 std::vector<ConversionTrace> traces;

47

48 void add(ConversionTrace trace) {

49 traces.push_back(std::move(trace));

50 }

51 void clear() {

52 traces.clear();

53 }

54};

55

56ConversionCost estimate_conversion_cost(ConversionKind kind, std::uint64_t bytes_copied);

57bool conversion_allowed(ConversionPolicy policy, ConversionKind kind);

58

59} // namespace simaai::neat

---

Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.1.