diff options
author | Dimitry Andric <dim@FreeBSD.org> | 2023-09-02 21:17:18 +0000 |
---|---|---|
committer | Dimitry Andric <dim@FreeBSD.org> | 2023-12-08 17:34:50 +0000 |
commit | 06c3fb2749bda94cb5201f81ffdb8fa6c3161b2e (patch) | |
tree | 62f873df87c7c675557a179e0c4c83fe9f3087bc /contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp | |
parent | cf037972ea8863e2bab7461d77345367d2c1e054 (diff) | |
parent | 7fa27ce4a07f19b07799a767fc29416f3b625afb (diff) |
Diffstat (limited to 'contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp')
-rw-r--r-- | contrib/llvm-project/llvm/lib/Analysis/TensorSpec.cpp | 21 |
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 |