Home
last modified time | relevance | path

Searched defs:atom_c (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/angle/
Dangle_energy_impl.cu22 … const int *atom_a, const int *atom_b, const int *atom_c, const float *angle_k, in AngleEnergyKernel()
51const int *atom_c, const float *angle_k, const float *angle_theta0, float *ene, cudaStream_t strea… in AngleEnergy()
Dangle_atom_energy_impl.cu22 … const int *atom_a, const int *atom_b, const int *atom_c, const float *angle_k, in AngleAtomEnergyKernel()
51 const int *atom_a, const int *atom_b, const int *atom_c, const float *angle_k, in AngleAtomEnergy()
Dangle_force_impl.cu22 … const int *atom_a, const int *atom_b, const int *atom_c, const float *angle_k, in AngleForceKernel()
70 …const int *atom_b, const int *atom_c, const float *angle_k, const float *angle_theta0, float *frc_… in AngleForce()
Dangle_force_with_atom_energy_impl.cu23const int *atom_c, const float *angle_k, const float *angle_theta0, in AngleForceWithAtomEnergyKernel()
74 … const int *atom_a, const int *atom_b, const int *atom_c, const float *angle_k, in AngleForceWithAtomEnergy()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/dihedral/
Ddihedral_energy_impl.cu22 … const int *atom_a, const int *atom_b, const int *atom_c, const int *atom_d, in DihedralEnergyKernel()
67 … const int *atom_b, const int *atom_c, const int *atom_d, const int *ipn, const float *pk, in DihedralEnergy()
Ddihedral_atom_energy_impl.cu22 … const VECTOR *scaler, const int *atom_a, const int *atom_b, const int *atom_c, in DihedralAtomEnergyKernel()
67 … const int *atom_a, const int *atom_b, const int *atom_c, const int *atom_d, const int *ipn, in DihedralAtomEnergy()
Ddihedral_force_impl.cu22 … const int *atom_a, const int *atom_b, const int *atom_c, const int *atom_d, in DihedralForceKernel()
103 … const int *atom_a, const int *atom_b, const int *atom_c, const int *atom_d, const int *ipn, in DihedralForce()
Ddihedral_force_with_atom_energy_impl.cu23const int *atom_c, const int *atom_d, const int *ipn, const float *pk, in DihedralForceWithAtomEnergyKernel()
107 … const int *atom_a, const int *atom_b, const int *atom_c, const int *atom_d, in DihedralForceWithAtomEnergy()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/angle/
Dangle_energy_kernel.h67 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
Dangle_force_kernel.h67 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
Dangle_atom_energy_kernel.h67 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
Dangle_force_with_atom_energy_kernel.h67 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/dihedral/
Ddihedral_force_with_atom_energy_kernel.h75 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
Ddihedral_energy_kernel.h75 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
Ddihedral_atom_energy_kernel.h75 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
Ddihedral_force_kernel.h75 auto atom_c = GetDeviceAddress<const T1>(inputs, 4); in Launch() local
/third_party/mesa3d/src/panfrost/bifrost/
Dbi_schedule.c324 bi_instr *atom_c = bi_atom_c_return_i32(&b, in bi_lower_atom_c() local
343 bi_instr *atom_c = bi_atom_c1_return_i32(&b, in bi_lower_atom_c1() local