fix sdft bug

haozhihan · haozhihan · commit 5373bdc90581 · 2024-12-31T09:53:50.000Z
diff --git a/source/module_psi/psi.cpp b/source/module_psi/psi.cpp
@@ -49,7 +49,7 @@ template <typename T, typename Device>
 Psi<T, Device>::Psi(const int nk_in, const int nbd_in, const int nbs_in, const int* ngk_in, const bool k_first_in)
 {
     assert(nk_in > 0);
-    assert(nbd_in > 0);
+    assert(nbd_in >= 0); // 187_PW_SDFT_ALL_GPU && 187_PW_MD_SDFT_ALL_GPU
     assert(nbs_in > 0);
 
     this->k_first = k_first_in;

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ template <typename T, typename Device>`
`49`	`49`	`Psi<T, Device>::Psi(const int nk_in, const int nbd_in, const int nbs_in, const int* ngk_in, const bool k_first_in)`
`50`	`50`	`{`
`51`	`51`	`assert(nk_in > 0);`
`52`		`- assert(nbd_in > 0);`
	`52`	`+ assert(nbd_in >= 0); // 187_PW_SDFT_ALL_GPU && 187_PW_MD_SDFT_ALL_GPU`
`53`	`53`	`assert(nbs_in > 0);`
`54`	`54`
`55`	`55`	`this->k_first = k_first_in;`