fix: psi error

dyzheng · dyzheng · commit 88a20db49d0e · 2025-09-10T22:00:50.000+08:00
diff --git a/source/source_psi/psi.cpp b/source/source_psi/psi.cpp
@@ -301,7 +301,7 @@ const int* Psi<T, Device>::get_ngk_pointer() const
 }
 
 template <typename T, typename Device>
-const int& Psi<T, Device>::get_psi_bias() const
+const size_t& Psi<T, Device>::get_psi_bias() const
 {
     return this->psi_bias;
 }
diff --git a/source/source_psi/psi.h b/source/source_psi/psi.h
@@ -130,7 +130,7 @@ class Psi
     const Device* get_device() const;
 
     // return psi_bias
-    const int& get_psi_bias() const;
+    const size_t& get_psi_bias() const;
 
     const int& get_current_ngk() const;
 

Original file line number	Diff line number	Diff line change
`@@ -301,7 +301,7 @@ const int* Psi<T, Device>::get_ngk_pointer() const`
`301`	`301`	`}`
`302`	`302`
`303`	`303`	`template <typename T, typename Device>`
`304`		`-const int& Psi<T, Device>::get_psi_bias() const`
	`304`	`+const size_t& Psi<T, Device>::get_psi_bias() const`
`305`	`305`	`{`
`306`	`306`	`return this->psi_bias;`
`307`	`307`	`}`