DeePMD-kit’s documentation
DeePMD-kit is a package written in Python/C++, designed to minimize the effort required to build deep learning-based models of interatomic potential energy and force field and to perform molecular dynamics (MD). This brings new hopes to addressing the accuracy-versus-efficiency dilemma in molecular simulations. Applications of DeePMD-kit span from finite molecules to extended systems and from metallic systems to chemically bonded systems.
Important
The project DeePMD-kit is licensed under GNU LGPLv3.0. If you use this code in any future publications, please cite this using Han Wang, Linfeng Zhang, Jiequn Han, and Weinan E. “DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics.” Computer Physics Communications 228 (2018): 178-184.
Getting Started
Advanced
- 1. Installation
- 2. Data
- 3. Model
- 3.1. Overall
- 3.2. Descriptor
"se_e2_a" - 3.3. Descriptor
"se_e2_r" - 3.4. Descriptor
"se_e3" - 3.5. Descriptor
"se_atten" - 3.6. Descriptor
"hybrid" - 3.7. Determine
sel - 3.8. Fit energy
- 3.9. Fit
tensorlikeDipoleandPolarizability - 3.10. Type embedding approach
- 3.11. Descriptor
"se_a_mask" - 3.12. Deep potential long-range (DPLR)
- 3.13. Deep Potential - Range Correction (DPRc)
- 4. Training
- 5. Freeze and Compress
- 6. Test
- 7. Inference
- 8. Command line interface
- 9. Integrate with third-party packages
- 10. Use NVNMD
- 11. FAQs
- 11.1. How to tune Fitting/embedding-net size ?
- 11.2. How to control the parallelism of a job?
- 11.3. Do we need to set rcut < half boxsize?
- 11.4. How to set sel?
- 11.5. Installation
- 11.6. The temperature undulates violently during the early stages of MD
- 11.7. MD: cannot run LAMMPS after installing a new version of DeePMD-kit
- 11.8. Model compatibility
- 11.9. Why does a model have low precision?
Tutorial
Developer Guide
- Find DeePMD-kit C/C++ library from CMake
- Coding Conventions
- Create a model
- Atom Type Embedding
- Python API
- deepmd package
DeepEvalDeepPotential()DipoleChargeModifier- Subpackages
- deepmd.cluster package
- deepmd.descriptor package
DescriptorDescrptHybridDescrptLocFrameDescrptSeADescrptSeAEbdDescrptSeAEfDescrptSeAEfLowerDescrptSeAMaskDescrptSeAttenDescrptSeRDescrptSeT- Submodules
- deepmd.descriptor.descriptor module
- deepmd.descriptor.hybrid module
- deepmd.descriptor.loc_frame module
- deepmd.descriptor.se module
- deepmd.descriptor.se_a module
- deepmd.descriptor.se_a_ebd module
- deepmd.descriptor.se_a_ef module
- deepmd.descriptor.se_a_mask module
- deepmd.descriptor.se_atten module
- deepmd.descriptor.se_r module
- deepmd.descriptor.se_t module
- deepmd.entrypoints package
compress()config()convert()doc_train_input()freeze()make_model_devi()neighbor_stat()test()train_dp()transfer()- Submodules
- deepmd.entrypoints.compress module
- deepmd.entrypoints.config module
- deepmd.entrypoints.convert module
- deepmd.entrypoints.doc module
- deepmd.entrypoints.freeze module
- deepmd.entrypoints.ipi module
- deepmd.entrypoints.main module
- deepmd.entrypoints.neighbor_stat module
- deepmd.entrypoints.test module
- deepmd.entrypoints.train module
- deepmd.entrypoints.transfer module
- deepmd.fit package
- deepmd.infer package
DeepDipoleDeepEvalDeepGlobalPolarDeepPolarDeepPotDeepPotential()DeepWFCDipoleChargeModifierEwaldRecpcalc_model_devi()- Submodules
- deepmd.infer.data_modifier module
- deepmd.infer.deep_dipole module
- deepmd.infer.deep_eval module
- deepmd.infer.deep_polar module
- deepmd.infer.deep_pot module
- deepmd.infer.deep_tensor module
- deepmd.infer.deep_wfc module
- deepmd.infer.ewald_recp module
- deepmd.infer.model_devi module
- deepmd.loggers package
- deepmd.loss package
- deepmd.model package
- deepmd.nvnmd package
- deepmd.op package
- deepmd.train package
- deepmd.utils package
DeepmdDataDeepmdDataSystemLearningRateExpPairTabPluginPluginVariant- Submodules
- deepmd.utils.argcheck module
- deepmd.utils.batch_size module
- deepmd.utils.compat module
- deepmd.utils.convert module
- deepmd.utils.data module
- deepmd.utils.data_system module
- deepmd.utils.errors module
- deepmd.utils.finetune module
- deepmd.utils.graph module
- deepmd.utils.learning_rate module
- deepmd.utils.multi_init module
- deepmd.utils.neighbor_stat module
- deepmd.utils.network module
- deepmd.utils.pair_tab module
- deepmd.utils.parallel_op module
- deepmd.utils.path module
- deepmd.utils.plugin module
- deepmd.utils.random module
- deepmd.utils.sess module
- deepmd.utils.tabulate module
- deepmd.utils.type_embed module
- deepmd.utils.weight_avg module
- Submodules
- deepmd.calculator module
- deepmd.common module
- deepmd.env module
- deepmd.lmp module
- deepmd package
- OP API
- op_module
AddFltNvnmd()CopyFltNvnmd()Descrpt()DescrptNorot()DescrptSeA()DescrptSeAEf()DescrptSeAEfPara()DescrptSeAEfVert()DescrptSeAMask()DescrptSeR()DotmulFltNvnmd()EwaldRecp()FltNvnmd()Gelu()GeluCustom()GeluGrad()GeluGradCustom()GeluGradGrad()GeluGradGradCustom()MapAparam()MapFltNvnmd()MatmulFitnetNvnmd()MatmulFlt2fixNvnmd()MatmulFltNvnmd()MulFltNvnmd()NeighborStat()PairTab()ParallelProdForceSeA()ProdEnvMatA()ProdEnvMatAMix()ProdEnvMatANvnmdQuantize()ProdEnvMatR()ProdForce()ProdForceNorot()ProdForceSeA()ProdForceSeAMask()ProdForceSeR()ProdVirial()ProdVirialNorot()ProdVirialSeA()ProdVirialSeR()QuantizeNvnmd()SoftMinForce()SoftMinSwitch()SoftMinVirial()TabulateFusion()TabulateFusionGrad()TabulateFusionGradGrad()TabulateFusionSeA()TabulateFusionSeAGrad()TabulateFusionSeAGradGrad()TabulateFusionSeR()TabulateFusionSeRGrad()TabulateFusionSeRGradGrad()TabulateFusionSeT()TabulateFusionSeTGrad()TabulateFusionSeTGradGrad()Tanh4FltNvnmd()UnaggregatedDy2Dx()UnaggregatedDy2DxS()UnaggregatedDyDx()UnaggregatedDyDxS()add_flt_nvnmd()copy_flt_nvnmd()descrpt()descrpt_norot()descrpt_se_a()descrpt_se_a_ef()descrpt_se_a_ef_para()descrpt_se_a_ef_vert()descrpt_se_a_mask()descrpt_se_r()dotmul_flt_nvnmd()ewald_recp()flt_nvnmd()gelu()gelu_custom()gelu_grad()gelu_grad_custom()gelu_grad_grad()gelu_grad_grad_custom()map_aparam()map_flt_nvnmd()matmul_fitnet_nvnmd()matmul_flt2fix_nvnmd()matmul_flt_nvnmd()mul_flt_nvnmd()neighbor_stat()pair_tab()parallel_prod_force_se_a()prod_env_mat_a()prod_env_mat_a_mix()prod_env_mat_a_nvnmd_quantize()prod_env_mat_r()prod_force()prod_force_norot()prod_force_se_a()prod_force_se_a_mask()prod_force_se_r()prod_virial()prod_virial_norot()prod_virial_se_a()prod_virial_se_r()quantize_nvnmd()soft_min_force()soft_min_switch()soft_min_virial()tabulate_fusion()tabulate_fusion_grad()tabulate_fusion_grad_grad()tabulate_fusion_se_a()tabulate_fusion_se_a_grad()tabulate_fusion_se_a_grad_grad()tabulate_fusion_se_r()tabulate_fusion_se_r_grad()tabulate_fusion_se_r_grad_grad()tabulate_fusion_se_t()tabulate_fusion_se_t_grad()tabulate_fusion_se_t_grad_grad()tanh4_flt_nvnmd()unaggregated_dy2_dx()unaggregated_dy2_dx_s()unaggregated_dy_dx()unaggregated_dy_dx_s()
- op_grads_module
ProdForceGrad()ProdForceSeAGrad()ProdForceSeAMaskGrad()ProdForceSeRGrad()ProdVirialGrad()ProdVirialSeAGrad()ProdVirialSeRGrad()SoftMinForceGrad()SoftMinVirialGrad()prod_force_grad()prod_force_se_a_grad()prod_force_se_a_mask_grad()prod_force_se_r_grad()prod_virial_grad()prod_virial_se_a_grad()prod_virial_se_r_grad()soft_min_force_grad()soft_min_virial_grad()
- op_module
- C++ API
- Class Hierarchy
- File Hierarchy
- Full API
- Namespaces
- Classes and Structs
- Functions
- Function deepmd::check_status
- Function deepmd::convert_pbtxt_to_pb
- Function deepmd::get_env_nthreads
- Function deepmd::load_op_library
- Function deepmd::model_compatable
- Function deepmd::name_prefix
- Function deepmd::print_summary
- Function deepmd::read_file_to_string
- Template Function deepmd::select_by_type
- Template Function deepmd::select_map(std::vector<VT>&, const std::vector<VT>&, const std::vector<int>&, const int&, const int&, const int&, const int&)
- Template Function deepmd::select_map(typename std::vector<VT>::iterator, const typename std::vector<VT>::const_iterator, const std::vector<int>&, const int&, const int&, const int&, const int&)
- Template Function deepmd::select_map_inv(std::vector<VT>&, const std::vector<VT>&, const std::vector<int>&, const int&)
- Template Function deepmd::select_map_inv(typename std::vector<VT>::iterator, const typename std::vector<VT>::const_iterator, const std::vector<int>&, const int&)
- Template Function deepmd::select_real_atoms
- Function deepmd::session_get_dtype
- Template Function deepmd::session_get_scalar
- Template Function deepmd::session_get_vector
- Template Function deepmd::session_input_tensors(std::vector<std::pair<std::string, tensorflow::Tensor>>&, const std::vector<VALUETYPE>&, const int&, const std::vector<int>&, const std::vector<VALUETYPE>&, const double&, const std::vector<VALUETYPE>&, const std::vector<VALUETYPE>&, const deepmd::AtomMap&, const std::string)
- Template Function deepmd::session_input_tensors(std::vector<std::pair<std::string, tensorflow::Tensor>>&, const std::vector<VALUETYPE>&, const int&, const std::vector<int>&, const std::vector<VALUETYPE>&, InputNlist&, const std::vector<VALUETYPE>&, const std::vector<VALUETYPE>&, const deepmd::AtomMap&, const int, const int, const std::string)
- Template Function deepmd::session_input_tensors_mixed_type
- Typedefs
- C API
- Class Hierarchy
- File Hierarchy
- Full API
- Namespaces
- Classes and Structs
- Functions
- Template Function _DP_DeepPotCompute
- Specialized Template Function _DP_DeepPotCompute< double >
- Specialized Template Function _DP_DeepPotCompute< float >
- Template Function _DP_DeepPotComputeMixedType
- Specialized Template Function _DP_DeepPotComputeMixedType< double >
- Specialized Template Function _DP_DeepPotComputeMixedType< float >
- Template Function _DP_DeepPotComputeNList
- Specialized Template Function _DP_DeepPotComputeNList< double >
- Specialized Template Function _DP_DeepPotComputeNList< float >
- Template Function _DP_DeepPotModelDeviComputeNList
- Specialized Template Function _DP_DeepPotModelDeviComputeNList< double >
- Specialized Template Function _DP_DeepPotModelDeviComputeNList< float >
- Template Function _DP_DeepTensorCompute
- Specialized Template Function _DP_DeepTensorCompute< double >
- Specialized Template Function _DP_DeepTensorCompute< float >
- Template Function _DP_DeepTensorComputeNList
- Specialized Template Function _DP_DeepTensorComputeNList< double >
- Specialized Template Function _DP_DeepTensorComputeNList< float >
- Template Function _DP_DeepTensorComputeTensor
- Specialized Template Function _DP_DeepTensorComputeTensor< double >
- Specialized Template Function _DP_DeepTensorComputeTensor< float >
- Template Function _DP_DeepTensorComputeTensorNList
- Specialized Template Function _DP_DeepTensorComputeTensorNList< double >
- Specialized Template Function _DP_DeepTensorComputeTensorNList< float >
- Template Function _DP_DipoleChargeModifierComputeNList
- Specialized Template Function _DP_DipoleChargeModifierComputeNList< double >
- Specialized Template Function _DP_DipoleChargeModifierComputeNList< float >
- Function _DP_Get_Energy_Pointer(std::vector<double>&, const int)
- Function _DP_Get_Energy_Pointer(double&, const int)
- Function deepmd::hpp::convert_nlist
- Function deepmd::hpp::convert_pbtxt_to_pb
- Function deepmd::hpp::read_file_to_string
- Function DP_ConvertPbtxtToPb
- Function DP_DeepPotCheckOK
- Function DP_DeepPotCompute
- Function DP_DeepPotCompute2
- Function DP_DeepPotComputef
- Function DP_DeepPotComputef2
- Function DP_DeepPotComputeMixedType
- Function DP_DeepPotComputeMixedTypef
- Function DP_DeepPotComputeNList
- Function DP_DeepPotComputeNList2
- Function DP_DeepPotComputeNListf
- Function DP_DeepPotComputeNListf2
- Function DP_DeepPotGetCutoff
- Function DP_DeepPotGetDimAParam
- Function DP_DeepPotGetDimFParam
- Function DP_DeepPotGetNumbTypes
- Function DP_DeepPotGetTypeMap
- Function DP_DeepPotModelDeviCheckOK
- Function DP_DeepPotModelDeviComputeNList
- Function DP_DeepPotModelDeviComputeNListf
- Function DP_DeepPotModelDeviGetCutoff
- Function DP_DeepPotModelDeviGetNumbTypes
- Function DP_DeepTensorCheckOK
- Function DP_DeepTensorCompute
- Function DP_DeepTensorComputef
- Function DP_DeepTensorComputeNList
- Function DP_DeepTensorComputeNListf
- Function DP_DeepTensorComputeTensor
- Function DP_DeepTensorComputeTensorf
- Function DP_DeepTensorComputeTensorNList
- Function DP_DeepTensorComputeTensorNListf
- Function DP_DeepTensorGetCutoff
- Function DP_DeepTensorGetNumbSelTypes
- Function DP_DeepTensorGetNumbTypes
- Function DP_DeepTensorGetOutputDim
- Function DP_DeepTensorGetSelTypes
- Function DP_DipoleChargeModifierCheckOK
- Function DP_DipoleChargeModifierComputeNList
- Function DP_DipoleChargeModifierComputeNListf
- Function DP_DipoleChargeModifierGetCutoff
- Function DP_DipoleChargeModifierGetNumbSelTypes
- Function DP_DipoleChargeModifierGetNumbTypes
- Function DP_DipoleChargeModifierGetSelTypes
- Function DP_NewDeepPot
- Function DP_NewDeepPotModelDevi
- Function DP_NewDeepPotWithParam
- Function DP_NewDeepTensor
- Function DP_NewDeepTensorWithParam
- Function DP_NewDipoleChargeModifier
- Function DP_NewDipoleChargeModifierWithParam
- Function DP_NewNlist
- Function DP_NlistCheckOK
- Function DP_PrintSummary
- Function DP_ReadFileToChar
- Defines
- Typedefs
- Core API
- Class Hierarchy
- File Hierarchy
- Full API
- Namespaces
- Classes and Structs
- Struct deepmd_exception
- Struct deepmd_exception_oom
- Template Struct EwaldParameters
- Struct InputNlist
- Template Struct Region
- Template Struct DescrptSeRGPUExecuteFunctor
- Template Struct GeluGPUExecuteFunctor
- Template Struct GeluGradGPUExecuteFunctor
- Template Struct GeluGradGradGPUExecuteFunctor
- Template Struct ProdForceSeAGPUExecuteFunctor
- Template Struct ProdForceSeRGPUExecuteFunctor
- Template Struct ProdVirialSeAGPUExecuteFunctor
- Template Struct ProdVirialSeRGPUExecuteFunctor
- Template Struct TabulateCheckerGPUExecuteFunctor
- Template Struct TabulateFusionGPUExecuteFunctor
- Template Struct TabulateFusionGradGPUExecuteFunctor
- Template Class SimulationRegion
- Unions
- Functions
- Template Function add_flt_nvnmd
- Function build_nlist(std::vector<std::vector<int>>&, std::vector<std::vector<int>>&, const std::vector<double>&, const int&, const double&, const double&, const std::vector<int>&, const std::vector<int>&, const std::vector<int>&, const std::vector<int>&, const SimulationRegion<double>&, const std::vector<int>&)
- Function build_nlist(std::vector<std::vector<int>>&, std::vector<std::vector<int>>&, const std::vector<double>&, const double&, const double&, const std::vector<int>&, const SimulationRegion<double>&)
- Function build_nlist(std::vector<std::vector<int>>&, std::vector<std::vector<int>>&, const std::vector<double>&, const std::vector<int>&, const std::vector<int>&, const double&, const double&, const std::vector<int>&, const SimulationRegion<double>&)
- Function build_nlist(std::vector<std::vector<int>>&, std::vector<std::vector<int>>&, const std::vector<double>&, const double&, const double&, const SimulationRegion<double> *)
- Function compute_descriptor(std::vector<double>&, std::vector<double>&, std::vector<double>&, const std::vector<double>&, const int&, const std::vector<int>&, const SimulationRegion<double>&, const bool&, const int&, const std::vector<int>&, const std::vector<int>&, const std::vector<int>&, const std::vector<int>&, const int, const int, const int, const int)
- Function compute_descriptor(std::vector<double>&, std::vector<double>&, std::vector<double>&, std::vector<double>&, std::vector<double>&, std::vector<double>&, std::vector<double>&, const std::vector<double>&, const int&, const std::vector<int>&, const SimulationRegion<double>&, const bool&, const int&, const std::vector<int>&, const std::vector<int>&, const std::vector<int>&, const std::vector<int>&, const int, const int, const int, const int)
- Function compute_descriptor_se_a_ef_para
- Function compute_descriptor_se_a_ef_vert
- Function compute_descriptor_se_a_extf
- Function compute_dRdT
- Function compute_dRdT_1
- Function compute_dRdT_2
- Function copy_coord
- Template Function deepmd::build_nlist_cpu
- Template Function deepmd::build_nlist_gpu
- Template Function deepmd::compute_cell_info
- Function deepmd::convert_nlist
- Function deepmd::convert_nlist_gpu_device
- Template Function deepmd::convert_to_inter_cpu
- Template Function deepmd::convert_to_inter_gpu
- Template Function deepmd::convert_to_phys_cpu
- Template Function deepmd::convert_to_phys_gpu
- Template Function deepmd::copy_coord_cpu
- Template Function deepmd::copy_coord_gpu
- Function deepmd::cos_switch(const double&, const double&, const double&)
- Function deepmd::cos_switch(double&, double&, const double&, const double&, const double&)
- Template Function deepmd::cprod
- Function deepmd::cum_sum
- Template Function deepmd::delete_device_memory
- Template Function deepmd::dot1
- Template Function deepmd::dot2
- Template Function deepmd::dot3
- Template Function deepmd::dot4
- Template Function deepmd::dotmv3
- Function deepmd::DPGetDeviceCount
- Function deepmd::DPSetDevice
- Template Function deepmd::env_mat_a_cpu
- Template Function deepmd::env_mat_a_nvnmd_quantize_cpu
- Function deepmd::env_mat_nbor_update
- Template Function deepmd::env_mat_r_cpu
- Template Function deepmd::ewald_recp
- Function deepmd::filter_ftype_gpu_cuda
- Template Function deepmd::format_nbor_list_gpu_cuda
- Template Function deepmd::format_nlist_cpu
- Function deepmd::free_nlist_gpu_device
- Template Function deepmd::gelu_cpu
- Template Function deepmd::gelu_gpu_cuda
- Template Function deepmd::gelu_grad_cpu
- Template Function deepmd::gelu_grad_gpu_cuda
- Template Function deepmd::gelu_grad_grad_cpu
- Template Function deepmd::gelu_grad_grad_gpu_cuda
- Template Function deepmd::init_region_cpu
- Template Function deepmd::invsqrt
- Specialized Template Function deepmd::invsqrt< double >
- Specialized Template Function deepmd::invsqrt< float >
- Template Function deepmd::malloc_device_memory(FPTYPE *&, const std::vector<FPTYPE>&)
- Template Function deepmd::malloc_device_memory(FPTYPE *&, const int)
- Template Function deepmd::malloc_device_memory(FPTYPE *&, std::vector<FPTYPE>&)
- Template Function deepmd::malloc_device_memory_sync(FPTYPE *&, const std::vector<FPTYPE>&)
- Template Function deepmd::malloc_device_memory_sync(FPTYPE *&, const FPTYPE *, const int)
- Template Function deepmd::malloc_device_memory_sync(FPTYPE *&, std::vector<FPTYPE>&)
- Template Function deepmd::map_aparam_cpu
- Function deepmd::max_numneigh
- Template Function deepmd::memcpy_device_to_host(const FPTYPE *, std::vector<FPTYPE>&)
- Template Function deepmd::memcpy_device_to_host(const FPTYPE *, FPTYPE *, const int)
- Template Function deepmd::memcpy_device_to_host(FPTYPE *, std::vector<FPTYPE>&)
- Template Function deepmd::memcpy_host_to_device(FPTYPE *, const std::vector<FPTYPE>&)
- Template Function deepmd::memcpy_host_to_device(FPTYPE *, const FPTYPE *, const int)
- Template Function deepmd::memcpy_host_to_device(FPTYPE *, std::vector<FPTYPE>&)
- Template Function deepmd::memset_device_memory
- Template Function deepmd::normalize_coord_cpu
- Template Function deepmd::normalize_coord_gpu
- Template Function deepmd::pair_tab_cpu
- Template Function deepmd::prod_env_mat_a_cpu
- Template Function deepmd::prod_env_mat_a_gpu_cuda
- Template Function deepmd::prod_env_mat_a_nvnmd_quantize_cpu
- Template Function deepmd::prod_env_mat_r_cpu
- Template Function deepmd::prod_env_mat_r_gpu_cuda
- Template Function deepmd::prod_force_a_cpu
- Template Function deepmd::prod_force_a_gpu_cuda
- Template Function deepmd::prod_force_grad_a_cpu
- Template Function deepmd::prod_force_grad_a_gpu_cuda
- Template Function deepmd::prod_force_grad_r_cpu
- Template Function deepmd::prod_force_grad_r_gpu_cuda
- Template Function deepmd::prod_force_r_cpu
- Template Function deepmd::prod_force_r_gpu_cuda
- Template Function deepmd::prod_virial_a_cpu
- Template Function deepmd::prod_virial_a_gpu_cuda
- Template Function deepmd::prod_virial_grad_a_cpu
- Template Function deepmd::prod_virial_grad_a_gpu_cuda
- Template Function deepmd::prod_virial_grad_r_cpu
- Template Function deepmd::prod_virial_grad_r_gpu_cuda
- Template Function deepmd::prod_virial_r_cpu
- Template Function deepmd::prod_virial_r_gpu_cuda
- Template Function deepmd::soft_min_switch_cpu
- Template Function deepmd::soft_min_switch_force_cpu
- Template Function deepmd::soft_min_switch_force_grad_cpu
- Template Function deepmd::soft_min_switch_virial_cpu
- Template Function deepmd::soft_min_switch_virial_grad_cpu
- Function deepmd::spline3_switch
- Template Function deepmd::spline5_switch
- Template Function deepmd::tabulate_fusion_se_a_cpu
- Template Function deepmd::tabulate_fusion_se_a_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_a_grad_cpu
- Template Function deepmd::tabulate_fusion_se_a_grad_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_a_grad_grad_cpu
- Template Function deepmd::tabulate_fusion_se_a_grad_grad_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_r_cpu
- Template Function deepmd::tabulate_fusion_se_r_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_r_grad_cpu
- Template Function deepmd::tabulate_fusion_se_r_grad_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_r_grad_grad_cpu
- Template Function deepmd::tabulate_fusion_se_r_grad_grad_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_t_cpu
- Template Function deepmd::tabulate_fusion_se_t_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_t_grad_cpu
- Template Function deepmd::tabulate_fusion_se_t_grad_gpu_cuda
- Template Function deepmd::tabulate_fusion_se_t_grad_grad_cpu
- Template Function deepmd::tabulate_fusion_se_t_grad_grad_gpu_cuda
- Template Function deepmd::test_encoding_decoding_nbor_info_gpu_cuda
- Function deepmd::use_nei_info_cpu
- Function deepmd::use_nei_info_gpu
- Function deepmd::use_nlist_map
- Template Function deepmd::volume_cpu
- Template Function deepmd::volume_gpu
- Template Function dotmul_flt_nvnmd
- Function DPAssert(cudaError_t, const char *, int, bool)
- Function DPAssert(hipError_t, const char *, int, bool)
- Function env_mat_a
- Function env_mat_r
- Template Function find_max_expo(int64_t&, T *, int64_t)
- Template Function find_max_expo(int64_t&, T *, int64_t, int64_t)
- Template Function format_nlist_i_cpu
- Function format_nlist_i_fill_a
- Template Function mul_flt_nvnmd
- Function nborAssert(cudaError_t, const char *, int, bool)
- Function nborAssert(hipError_t, const char *, int, bool)
- Function omp_get_num_threads
- Function omp_get_thread_num
- Template Function split_flt
- Variables
- Defines
- Typedefs
Project Details