Merge branch 'deepmodeling:develop' into develop

Author: QuantumMisaka

.github/workflows/cuda.yml

@@ -37,7 +37,22 @@ jobs:
           cmake --build build -j4
           cmake --install build
 
-      - name: Test
+      - name: Test 11_PW_GPU
         run: |
-          cd tests/integrate
-          bash Autotest.sh -n 2 -f CASES_GPU.txt
+          cd tests/11_PW_GPU
+          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt
+
+      - name: Test 12_NAO_Gamma_GPU
+        run: |
+          cd tests/12_NAO_Gamma_GPU
+          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt
+
+      - name: Test 13_NAO_multik_GPU
+        run: |
+          cd tests/13_NAO_multik_GPU
+          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt
+
+      - name: Test 16_SDFT_GPU
+        run: |
+          cd tests/16_SDFT_GPU
+          bash ../integrate/Autotest.sh -n 2 -f CASES_GPU.txt

.github/workflows/test.yml

@@ -48,9 +48,88 @@ jobs:
         run: |
           cmake --build build -j8
           cmake --install build
-      - name: Test
+
+      - name: Integrated Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R integrated_test"
+
+      - name: 01_PW Test
         env:
           GTEST_COLOR: 'yes'
           OMP_NUM_THREADS: '2'
         run: |
-          cmake --build build --target test ARGS="-V --timeout 1700"
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 01_PW"
+
+      - name: 02_NAO_Gamma Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 02_NAO_Gamma"
+
+      - name: 03_NAO_multik Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 03_NAO_multik"
+
+      - name: 04_LJ_DP Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 04_LJ_DP"
+
+      - name: 05_rtTDDFT Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 05_rtTDDFT"
+
+      - name: 06_SDFT Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 06_SDFT"
+
+      - name: 07_OFDFT Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 07_OFDFT"
+
+      - name: 08_EXX Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 08_EXX"
+
+      - name: 09_DeePKS Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 09_DeePKS"
+
+      - name: 10_others Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -R 10_others"
+
+      - name: Unit Test
+        env:
+          GTEST_COLOR: 'yes'
+          OMP_NUM_THREADS: '2'
+        run: |
+          cmake --build build --target test ARGS="-V --timeout 1700 -E 'integrate_test|01_PW|02_NAO_Gamma|03_NAO_multik|04_LJ_DP|05_rtTDDFT|06_SDFT|07_OFDFT|08_EXX|09_DeePKS|10_others|11_PW_GPU|16_SDFT_GPU'"
+

CMakeLists.txt

@@ -190,6 +190,7 @@ find_program(CCACHE ccache)
 if(CCACHE)
   set(CMAKE_CXX_COMPILER_LAUNCHER ${CCACHE} ${CMAKE_CXX_COMPILER_LAUNCHER})
   set(CMAKE_C_COMPILER_LAUNCHER ${CCACHE} ${CMAKE_C_COMPILER_LAUNCHER})
+  set(CMAKE_CUDA_COMPILER_LAUNCHER ${CCACHE} ${CMAKE_CUDA_COMPILER_LAUNCHER})
 endif()
 
 # Choose build type from: Debug Release RelWithDebInfo MinSizeRel Select
@@ -797,4 +798,4 @@ install(PROGRAMS ${ABACUS_BIN_PATH}
 
 if(ENABLE_COVERAGE)
   coverage_evaluate()
-endif()
+endif()

docs/advanced/input_files/input-main.md

@@ -1645,10 +1645,13 @@ These variables are used to control the output of properties.
 ### out_dm
 
 - **Type**: Boolean
-- **Availability**: Numerical atomic orbital basis (gamma-only algorithm)
+- **Availability**: Numerical atomic orbital basis
 - **Description**: Whether to output the density matrix of localized orbitals into files in the folder `OUT.${suffix}`. The files are named as:
-  - nspin = 1: SPIN1_DM;
-  - nspin = 2: SPIN1_DM, and SPIN2_DM.
+  - For gamma only case:
+    - nspin = 1: SPIN1_DM;
+    - nspin = 2: SPIN1_DM, and SPIN2_DM.
+  - For multi-k points case:
+    - SPIN\*_K\*_DM, where \* stands for index of spin and kpoints;
 - **Default**: False
 
 ### out_dm1

source/module_basis/module_ao/ORB_gaunt_table.cpp

@@ -162,16 +162,6 @@ int ORB_gaunt_table::get_lm_index(
 }
 
 
-/**********************//**
-  * Rasch and Yu's Method
-  ***********************/
-///total pointers
-//int ORB_gaunt_table::P_EL(const int& L)
-//{
-//	return (L+1) * (L+2) * (L+3) * (L+4) / 24;
-//}
-
-
 ///effective pointers
 int ORB_gaunt_table::EP_EL(const int& L)
 {
@@ -201,13 +191,7 @@ void ORB_gaunt_table::init_Gaunt_CH(const int& Lmax)
 	ModuleBase::TITLE("ORB_gaunt_table","init_Gaunt_CH");
 	ModuleBase::timer::tick("ORB_gaunt_table","init_Gaunt_CH");
 
-//	assert(Lmax <= 6);			// Peize Lin delete 2016-08-26. why?
-
 	int L = 2*Lmax + 1;
-	
-	//int Np = this->P_EL(L);
-//	assert(Np <= 5000);			// Peize Lin delete 2016-08-26. why?
-
 	int Eff_Np = this->EP_EL(L);
 
 	ModuleBase::Memory::record("ORB::Gaunt_CH", sizeof(double) * Eff_Np * 30);
@@ -228,9 +212,6 @@ void ORB_gaunt_table::init_Gaunt_CH(const int& Lmax)
 					if((l_sum % 2 == 0) && (l2 + l3 >= l1))
 					{
 						int uplmt_m2 = l1 - m3 > l2 ? l2 : l1 - m3;
-						//int dim = l2 + uplmt_m2 + 1;
-						
-//						assert(dim <= 30);			// Peize Lin delete 2016-08-26. why?
 
 						int ic2 = 0;
 						for(int m2 = -l2; m2 <= uplmt_m2; m2++)
@@ -239,9 +220,6 @@ void ORB_gaunt_table::init_Gaunt_CH(const int& Lmax)
 							int m1 = -m2 - m3;
 							assert(std::abs(m1) <= l1);
 
-							// Peize Lin delete assert 2016-08-26
-//							assert(ic1 < 5000);
-//							assert(ic2 < 30);
 							Gaunt_CH[ic1][ic2] = Calc_Gaunt_CH(l1, m1, l2, m2, l3, m3);
 							ic2++;
 						}
@@ -269,7 +247,6 @@ double ORB_gaunt_table::Calc_Gaunt_CH
 	const int& m3
 )
 {
-//	ModuleBase::TITLE("ORB_gaunt_table","Calc_Gaunt_CH");
 	ModuleBase::timer::tick("ORB_gaunt_table","Calc_Gaunt_CH");
 
 	double fac = sqrt((2*l1+1) * (2*l2+1) * (2*l3+1) / ModuleBase::FOUR_PI);
@@ -281,7 +258,8 @@ double ORB_gaunt_table::Calc_Gaunt_CH
 
 	double aux1 = sqrt(Fact(l1+m1) * Fact(l1-m1) * Fact(l2+m2) * Fact(l2-m2) * Fact(l3+m3) * Fact(l3-m3));
 
-	int kmin, kmax;
+	int kmin = 0;
+    int kmax = 0;
 
 	kmin = (l2-l3-m1) > (l1-l3+m2) ? (l2-l3-m1) : (l1-l3+m2);
 	kmin = kmin > 0 ? kmin : 0;
@@ -324,16 +302,14 @@ double ORB_gaunt_table::Get_Gaunt_CH
 
 	if( (m1 + m2 + m3) != 0) return 0.0;
 
-	int L1, M1, L2, M2, L3, M3;
-	
-	L1 = l1;
-	M1 = m1;
-	L2 = l2;
-	M2 = m2;
+    int L1 = l1;
+	int M1 = m1;
+	int L2 = l2;
+    int M2 = m2;
 	Swap(L1, M1, L2, M2);
 
-	L3 = l3;
-	M3 = m3;
+	int L3 = l3;
+	int M3 = m3;
 	Swap(L1, M1, L3, M3);
 
 	Swap(L2, M2, L3, M3);
@@ -345,19 +321,17 @@ double ORB_gaunt_table::Get_Gaunt_CH
 		M3 = -M3;
 	}
 
-	/*	
-	if(l1 == 2 && m1 == -1 && l2 == 2 && m2 == 2 && l3 == 2 && m3 == -1)
-	{
-		std::cout << L1 << " " << L2 << " " << L3 << std::endl;
-		std::cout << M1 << " " << M2 << " " << M3 <<std::endl;
-	}
-	*/
-	
 	int ic1 = index_func(L1, L2, L3, M3);
 	int ic2 = M2 + L2;
 
-	try{ return Gaunt_CH.at(ic1).at(ic2); }			// Peize Lin add 2016-08-26
-	catch( std::out_of_range ){ return 0; }
+	try
+	{ 
+		return Gaunt_CH.at(ic1).at(ic2); 
+	} // Peize Lin add 2016-08-26
+	catch( std::out_of_range )
+	{ 
+		return 0; 
+	}
 }
 
 
@@ -374,8 +348,6 @@ double ORB_gaunt_table::Get_Gaunt_SH
 	const int& mm3
 )
 {
-//	ModuleBase::TITLE("ORB_gaunt_table","Get_Gaunt_SH");
-	ModuleBase::timer::tick("ORB_gaunt_table","Get_Gaunt_SH");
 
 	//Tranform M index
 	int m1 = Index_M(mm1);
@@ -442,7 +414,6 @@ double ORB_gaunt_table::Get_Gaunt_SH
 		else return 0.0;
 	}
 
-	ModuleBase::timer::tick("ORB_gaunt_table","Get_Gaunt_SH");
 }
 
 

source/module_cell/print_cell.cpp

@@ -18,33 +18,61 @@ namespace unitcell
         // assert (direct || Coordinate == "Cartesian" || Coordinate == "Cartesian_angstrom"); // this line causes abort in unittest ReadAtomPositionsCACXY.
         // previously there are two if-statements, the first is `if(Coordinate == "Direct")` and the second is `if(Coordinate == "Cartesian" || Coordiante == "Cartesian_angstrom")`
         // however the Coordinate can also be value among Cartesian_angstrom_center_xy, Cartesian_angstrom_center_xz, Cartesian_angstrom_center_yz and Cartesian_angstrom_center_xyz
+
         // if Coordinate has value one of them, this print_tau will not print anything.
         std::regex pattern("Direct|Cartesian(_angstrom)?(_center_(xy|xz|yz|xyz))?");
         assert(std::regex_search(Coordinate, pattern));
         bool direct = (Coordinate == "Direct");
+
+        //----------------------
+        // print atom positions
+        //----------------------
         std::string table;
-        table += direct? "DIRECT COORDINATES\n": FmtCore::format(" CARTESIAN COORDINATES ( UNIT = %15.8f Bohr ).\n", lat0);
-        table += FmtCore::format("%8s%10s%10s%10s%8s%8s%8s%8s\n", "atom", "x", "y", "z", "mag", "vx", "vy", "vz");
+        table += direct? " DIRECT COORDINATES\n": FmtCore::format(" CARTESIAN COORDINATES ( UNIT = %15.8f Bohr ).\n", lat0);
+        table += FmtCore::format("%5s%19s%19s%19s%8s\n", "atom", "x", "y", "z", "mag");
         for(int it = 0; it < ntype; it++)
         {
             for (int ia = 0; ia < atoms[it].na; ia++)
             {
                 const double& x = direct? atoms[it].taud[ia].x: atoms[it].tau[ia].x;
                 const double& y = direct? atoms[it].taud[ia].y: atoms[it].tau[ia].y;
                 const double& z = direct? atoms[it].taud[ia].z: atoms[it].tau[ia].z;
-                table += FmtCore::format("%8s%10.3f%10.3f%10.3f%8.3f%8.3f%8.3f%8.3f\n", 
+                table += FmtCore::format("%5s%19.12f%19.12f%19.12f%8.4f\n", 
                                         atoms[it].label, 
                                         x, 
                                         y, 
                                         z, 
-                                        atoms[it].mag[ia], 
-                                        atoms[it].vel[ia].x, 
-                                        atoms[it].vel[ia].y, 
-                                        atoms[it].vel[ia].z);
+                                        atoms[it].mag[ia]); 
             }
         }
         table += "\n";
-        ofs << table << std::endl;
+        ofs << table; 
+
+
+        // print velocities
+        ofs << " ATOMIC VELOCITIES" << std::endl;
+        ofs << std::setprecision(12);
+        ofs << std::setw(5) << "atom" 
+            << std::setw(19) << "vx" 
+            << std::setw(19) << "vy" 
+            << std::setw(19) << "vz"
+            << std::endl;
+ 
+		for(int it = 0; it < ntype; it++)
+		{
+			for (int ia = 0; ia < atoms[it].na; ia++)
+			{
+                ofs << std::setw(5) << atoms[it].label;
+                ofs << " " << std::setw(18) << atoms[it].vel[ia].x;
+                ofs << " " << std::setw(18) << atoms[it].vel[ia].y;
+                ofs << " " << std::setw(18) << atoms[it].vel[ia].z;
+                ofs << std::endl;
+			}
+		}
+        ofs << std::endl;
+        ofs << std::setprecision(6); // return to 6, as original
+
+
         return;
     }
 

source/module_cell/test/unitcell_test.cpp

@@ -1022,11 +1022,10 @@ TEST_F(UcellTest, PrintTauDirect)
     ifs.open("print_tau_direct");
     std::string str((std::istreambuf_iterator<char>(ifs)), std::istreambuf_iterator<char>());
     EXPECT_THAT(str, testing::HasSubstr("DIRECT COORDINATES"));
-    EXPECT_THAT(str, testing::HasSubstr("       C     0.100     0.100     0.100   0.000   0.100   0.100   0.100"));
-    EXPECT_THAT(str, testing::HasSubstr("       H     0.150     0.150     0.150   0.000   0.100   0.100   0.100"));
+    EXPECT_THAT(str, testing::HasSubstr("    C     0.100000000000     0.100000000000     0.100000000000  0.0000"));
+    EXPECT_THAT(str, testing::HasSubstr("    H     0.150000000000     0.150000000000     0.150000000000  0.0000")); 
     ifs.close();
 
-    // remove the file
     remove("print_tau_direct");
 }
 
@@ -1047,8 +1046,8 @@ TEST_F(UcellTest, PrintTauCartesian)
     ifs.open("print_tau_Cartesian");
     std::string str((std::istreambuf_iterator<char>(ifs)), std::istreambuf_iterator<char>());
     EXPECT_THAT(str, testing::HasSubstr("CARTESIAN COORDINATES"));
-    EXPECT_THAT(str, testing::HasSubstr("       C     1.000     1.000     1.000   0.000   0.000   0.000   0.000"));
-    EXPECT_THAT(str, testing::HasSubstr("       H     1.500     1.500     1.500   0.000   0.000   0.000   0.000"));
+    EXPECT_THAT(str, testing::HasSubstr("    C     1.000000000000     1.000000000000     1.000000000000  0.0000"));
+    EXPECT_THAT(str, testing::HasSubstr("    H     1.500000000000     1.500000000000     1.500000000000  0.0000"));
     ifs.close();
 
     // remove the file

source/module_cell/test/unitcell_test_readpp.cpp

@@ -173,11 +173,8 @@ TEST_F(UcellDeathTest, ReadCellPPWarning4) {
     testing::internal::CaptureStdout();
     EXPECT_NO_THROW(elecstate::read_cell_pseudopots(pp_dir, ofs, *ucell));
     output = testing::internal::GetCapturedStdout();
-    EXPECT_THAT(output, testing::HasSubstr("dft_functional readin is: LDA"));
-    EXPECT_THAT(output,
-                testing::HasSubstr("dft_functional in pseudopot file is: PBE"));
-    EXPECT_THAT(output,
-                testing::HasSubstr("Please make sure this is what you need"));
+    EXPECT_THAT(output, testing::HasSubstr("DFT FUNC. (PSEUDO)   : PBE"));
+    EXPECT_THAT(output, testing::HasSubstr("DFT FUNC. (SET TO)   : LDA"));
 }
 
 TEST_F(UcellDeathTest, ReadCellPPWarning5) {