aboutsummaryrefslogtreecommitdiff
path: root/contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp
diff options
context:
space:
mode:
authorDimitry Andric <dim@FreeBSD.org>2023-09-02 21:17:18 +0000
committerDimitry Andric <dim@FreeBSD.org>2023-12-08 17:34:50 +0000
commit06c3fb2749bda94cb5201f81ffdb8fa6c3161b2e (patch)
tree62f873df87c7c675557a179e0c4c83fe9f3087bc /contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp
parentcf037972ea8863e2bab7461d77345367d2c1e054 (diff)
parent7fa27ce4a07f19b07799a767fc29416f3b625afb (diff)
Diffstat (limited to 'contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp')
-rw-r--r--contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp21
1 files changed, 21 insertions, 0 deletions
diff --git a/contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp b/contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp
index 4f7428ded85e..8dd1a054af88 100644
--- a/contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp
+++ b/contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp
@@ -10,8 +10,10 @@
// utils.
//
//===----------------------------------------------------------------------===//
+#include "llvm/ADT/STLExtras.h"
#include "llvm/Config/config.h"
+#include "llvm/ADT/StringExtras.h"
#include "llvm/ADT/Twine.h"
#include "llvm/Analysis/TensorSpec.h"
#include "llvm/Support/CommandLine.h"
@@ -102,4 +104,23 @@ std::optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx,
return std::nullopt;
}
+std::string tensorValueToString(const char *Buffer, const TensorSpec &Spec) {
+ switch (Spec.type()) {
+#define _IMR_DBG_PRINTER(T, N) \
+ case TensorType::N: { \
+ const T *TypedBuff = reinterpret_cast<const T *>(Buffer); \
+ auto R = llvm::make_range(TypedBuff, TypedBuff + Spec.getElementCount()); \
+ return llvm::join( \
+ llvm::map_range(R, [](T V) { return std::to_string(V); }), ","); \
+ }
+ SUPPORTED_TENSOR_TYPES(_IMR_DBG_PRINTER)
+#undef _IMR_DBG_PRINTER
+ case TensorType::Total:
+ case TensorType::Invalid:
+ llvm_unreachable("invalid tensor type");
+ }
+ // To appease warnings about not all control paths returning a value.
+ return "";
+}
+
} // namespace llvm