From 3ef34fb300e7f62bdb397c605ab6c3bd30682cf8 Mon Sep 17 00:00:00 2001
From: Tai Ly <tai.ly@arm.com>
Date: Tue, 4 Apr 2023 20:34:05 +0000
Subject: Add readFromNpyFile and writeToNpyFile for double data

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Icc023cbe6aa8843cc37d25e740bc6ce05bb7abb2
---
 src/numpy_utils.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'src/numpy_utils.cpp')

diff --git a/src/numpy_utils.cpp b/src/numpy_utils.cpp
index c770d45..123908a 100644
--- a/src/numpy_utils.cpp
+++ b/src/numpy_utils.cpp
@@ -46,6 +46,12 @@ NumpyUtilities::NPError NumpyUtilities::readFromNpyFile(const char* filename, co
     return readFromNpyFileCommon(filename, dtype_str, sizeof(float), elems, databuf, false);
 }
 
+NumpyUtilities::NPError NumpyUtilities::readFromNpyFile(const char* filename, const uint32_t elems, double* databuf)
+{
+    const char dtype_str[] = "'<f8'";
+    return readFromNpyFileCommon(filename, dtype_str, sizeof(double), elems, databuf, false);
+}
+
 NumpyUtilities::NPError
     NumpyUtilities::readFromNpyFile(const char* filename, const uint32_t elems, half_float::half* databuf)
 {
@@ -315,6 +321,20 @@ NumpyUtilities::NPError
     return writeToNpyFileCommon(filename, dtype_str, sizeof(float), shape, databuf, false);
 }
 
+NumpyUtilities::NPError
+    NumpyUtilities::writeToNpyFile(const char* filename, const uint32_t elems, const double* databuf)
+{
+    std::vector<int32_t> shape = { (int32_t)elems };
+    return writeToNpyFile(filename, shape, databuf);
+}
+
+NumpyUtilities::NPError
+    NumpyUtilities::writeToNpyFile(const char* filename, const std::vector<int32_t>& shape, const double* databuf)
+{
+    const char dtype_str[] = "'<f8'";
+    return writeToNpyFileCommon(filename, dtype_str, sizeof(double), shape, databuf, false);
+}
+
 NumpyUtilities::NPError NumpyUtilities::writeToNpyFile(const char* filename,
                                                        const std::vector<int32_t>& shape,
                                                        const half_float::half* databuf)
-- 
cgit v1.2.1