Small bug fix

Critsium-xy · Critsium-xy · commit eb7c145b3501 · 2025-01-16T11:09:01.000+08:00
diff --git a/source/module_base/module_device/memory_op.cpp b/source/module_base/module_device/memory_op.cpp
@@ -400,13 +400,13 @@ template struct delete_memory_op_mt<std::complex<double>, base_device::DEVICE_CP
 #endif
 
 template <typename FPTYPE>
-void resize_memory(FPTYPE* arr, base_device::AbacusDevice_t device_type)
+void resize_memory(FPTYPE* arr, const size_t size, base_device::AbacusDevice_t device_type)
 {
     if (device_type == base_device::AbacusDevice_t::CpuDevice){
-        resize_memory_op<FPTYPE, base_device::DEVICE_CPU>()(arr);
+        resize_memory_op<FPTYPE, base_device::DEVICE_CPU>()(arr, size);
     }
     else if (device_type == base_device::AbacusDevice_t::GpuDevice){
-        resize_memory_op<FPTYPE, base_device::DEVICE_GPU>()(arr);
+        resize_memory_op<FPTYPE, base_device::DEVICE_GPU>()(arr, size);
     }
 }
 
diff --git a/source/module_base/module_device/memory_op.h b/source/module_base/module_device/memory_op.h
@@ -93,7 +93,7 @@ struct delete_memory_op
 };
 
 template <typename FPTYPE>
-void resize_memory(FPTYPE* arr, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice);
+void resize_memory(FPTYPE* arr, const size_t size, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice);
 
 template <typename FPTYPE>
 void set_memory(FPTYPE* arr, const int var, const size_t size, base_device::AbacusDevice_t device_type = base_device::AbacusDevice_t::CpuDevice);

Original file line number	Diff line number	Diff line change
`@@ -400,13 +400,13 @@ template struct delete_memory_op_mt<std::complex<double>, base_device::DEVICE_CP`
`400`	`400`	`#endif`
`401`	`401`
`402`	`402`	`template <typename FPTYPE>`
`403`		`-void resize_memory(FPTYPE* arr, base_device::AbacusDevice_t device_type)`
	`403`	`+void resize_memory(FPTYPE* arr, const size_t size, base_device::AbacusDevice_t device_type)`
`404`	`404`	`{`
`405`	`405`	`if (device_type == base_device::AbacusDevice_t::CpuDevice){`
`406`		`- resize_memory_op<FPTYPE, base_device::DEVICE_CPU>()(arr);`
	`406`	`+ resize_memory_op<FPTYPE, base_device::DEVICE_CPU>()(arr, size);`
`407`	`407`	`}`
`408`	`408`	`else if (device_type == base_device::AbacusDevice_t::GpuDevice){`
`409`		`- resize_memory_op<FPTYPE, base_device::DEVICE_GPU>()(arr);`
	`409`	`+ resize_memory_op<FPTYPE, base_device::DEVICE_GPU>()(arr, size);`
`410`	`410`	`}`
`411`	`411`	`}`
`412`	`412`