From 4a1741aa2dd589eabaa7820aea567e1fe55bba1d Mon Sep 17 00:00:00 2001
From: klosax <131523366+klosax@users.noreply.github.com>
Date: Tue, 15 Aug 2023 19:56:19 +0200
Subject: [PATCH] gptneox-main.cpp : add tensor data layout

---
 gptneox-main.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/gptneox-main.cpp b/gptneox-main.cpp
index f336ee88b..aca8a5b0a 100644
--- a/gptneox-main.cpp
+++ b/gptneox-main.cpp
@@ -381,6 +381,8 @@ bool gpt_neox_model_load(const std::string & fname, gpt_neox_model & model, gpt2
         if (keyidx != -1) { fprintf(stdout, "%s: model architecture   = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
         keyidx = gguf_find_key(ggufctx, "general.file_type");
         if (keyidx != -1) { fprintf(stdout, "%s: model file type      = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
+        keyidx = gguf_find_key(ggufctx, "gptneox.tensor_data_layout");
+        if (keyidx != -1) { fprintf(stdout, "%s: model data layout    = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
         keyidx = gguf_find_key(ggufctx, "general.source.hugginface.repository");
         if (keyidx != -1) { fprintf(stdout, "%s: model source HF repo = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
     }