Fix 3

intel · Jun 13, 2024 · f6e6e64 · f6e6e64
1 parent 6a85769
commit f6e6e64
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/layers/rms_norm.cpp b/src/layers/rms_norm.cpp
@@ -70,7 +70,8 @@ void RmsNormImp<T>::setWeight(const float *w, const float *, int cols) {
 template <typename T>
 void RmsNormImp<T>::setWeight(const std::string &modelPath, const std::string &, int cols) {
     float weightBuf[cols];
-    loadWeight(modelPath, weightBuf, cols);
+    float *weiBuf = &weightBuf[0];
+    loadWeight(modelPath, weiBuf, cols);
     this->normSize = cols;
     this->setWeight(weightBuf, nullptr, cols);
 }

diff --git a/src/utils/weight_util.h b/src/utils/weight_util.h
@@ -153,7 +153,7 @@ int loadWeightWithConvert(T *ptr, int size, const std::string &filename, bool re
 }
 
 template <typename T>
-int loadWeight(std::string filename, T *ptr, int size, DataType w_type = DataType::unknown, bool required = true) {
+int loadWeight(std::string filename, T *&ptr, int size, DataType w_type = DataType::unknown, bool required = true) {
 
     // By default, read the config.ini configuration file
     // in the same directory as the model file to determine the data type of the file.