fake时更新unitsize

ztxz16 · Aug 13, 2024 · 7d1973a · 7d1973a
1 parent d3778c7
commit 7d1973a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/fastllm.cpp b/src/fastllm.cpp
@@ -289,6 +289,7 @@ namespace fastllm {
 
     void Data::FakeFrom(const Data &ori, size_t offset) {
         this->dataType = ori.dataType;
+        this->UpdateUnitSize();
         this->isFake = true;
         this->dataDevice = ori.dataDevice;
         if (this->dataDevice == DataDevice::CPU) {