Home
last modified time | relevance | path

Searched refs:VECTOR (Results 1 – 25 of 113) sorted by relevance

12345

/third_party/flutter/skia/src/sksl/
DSkSLByteCode.h21 #define VECTOR(name) name, name ## 2, name ## 3, name ## 4 macro
27 VECTOR(kAddI), enumerator
37 VECTOR(kCompareIEQ), enumerator
38 VECTOR(kCompareINEQ), enumerator
40 VECTOR(kCompareFGT), enumerator
41 VECTOR(kCompareFGTEQ), enumerator
42 VECTOR(kCompareFLT), enumerator
43 VECTOR(kCompareFLTEQ), enumerator
45 VECTOR(kCompareSGT), enumerator
46 VECTOR(kCompareSGTEQ), enumerator
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/simple_constrain/
Dconstrain_force_virial_impl.cu26 … const VECTOR *scaler, CONSTRAIN_PAIR *constrain_pair, in constrain_force_cycle_with_virial_update_kernel()
27 … const VECTOR *pair_dr, VECTOR *test_frc, in constrain_force_cycle_with_virial_update_kernel()
32 VECTOR dr0 = pair_dr[pair_i]; in constrain_force_cycle_with_virial_update_kernel()
33 …VECTOR dr = Get_Periodic_Displacement(uint_crd[cp.atom_i_serial], uint_crd[cp.atom_j_serial], scal… in constrain_force_cycle_with_virial_update_kernel()
36 VECTOR frc_lin = frc_abs * dr0; in constrain_force_cycle_with_virial_update_kernel()
50 … const UNSIGNED_INT_VECTOR *uint_crd, const VECTOR *scaler, in constrain_force_cycle_update_kernel()
51 … const CONSTRAIN_PAIR *constrain_pair, const VECTOR *pair_dr, in constrain_force_cycle_update_kernel()
52 VECTOR *test_frc) { in constrain_force_cycle_update_kernel()
57 VECTOR dr; in constrain_force_cycle_update_kernel()
59 VECTOR frc_lin; in constrain_force_cycle_update_kernel()
[all …]
Drefresh_crd_vel_impl.cu21 … float half_exp_gamma_plus_half, VECTOR *test_frc, float *mass_inverse, VECTOR *crd, in Refresh_Crd_Vel()
22 VECTOR *vel) { in Refresh_Crd_Vel()
25 VECTOR crd_lin = crd[atom_i]; in Refresh_Crd_Vel()
26 VECTOR frc_lin = test_frc[atom_i]; in Refresh_Crd_Vel()
27 VECTOR vel_lin = vel[atom_i]; in Refresh_Crd_Vel()
51 VECTOR *crd = reinterpret_cast<VECTOR *>(crd_f); in refreshcrdvel()
52 VECTOR *vel = reinterpret_cast<VECTOR *>(vel_f); in refreshcrdvel()
53 VECTOR *test_frc = reinterpret_cast<VECTOR *>(test_frc_f); in refreshcrdvel()
Dlast_crd_to_dr_impl.cu20 __global__ void Last_Crd_To_dr(const int constarin_pair_numbers, const VECTOR *atom_crd, in Last_Crd_To_dr()
21 … const VECTOR *quarter_crd_to_uint_crd_cof, const VECTOR *uint_dr_to_dr, in Last_Crd_To_dr()
22 CONSTRAIN_PAIR *constrain_pair, VECTOR *pair_dr) { in Last_Crd_To_dr()
30 VECTOR dr; in Last_Crd_To_dr()
64 const VECTOR *atom_crd = reinterpret_cast<const VECTOR *>(atom_crd_f); in lastcrdtodr()
65 …const VECTOR *quarter_crd_to_uint_crd_cof = reinterpret_cast<const VECTOR *>(quarter_crd_to_uint_c… in lastcrdtodr()
66 const VECTOR *uint_dr_to_dr = reinterpret_cast<const VECTOR *>(uint_dr_to_dr_f); in lastcrdtodr()
70 VECTOR *pair_dr = reinterpret_cast<VECTOR *>(pair_dr_f); in lastcrdtodr()
Drefresh_uint_crd_impl.cu20 __global__ void Refresh_Uint_Crd(int atom_numbers, const VECTOR *crd, const VECTOR *quarter_crd_to_… in Refresh_Uint_Crd()
21 … UNSIGNED_INT_VECTOR *uint_crd, const VECTOR *test_frc, const float *mass_inverse, in Refresh_Uint_Crd()
26 VECTOR crd_lin = crd[atom_i]; in Refresh_Uint_Crd()
27 VECTOR frc_lin = test_frc[atom_i]; in Refresh_Uint_Crd()
49 const VECTOR *crd = reinterpret_cast<const VECTOR *>(crd_f); in refreshuintcrd()
50 …const VECTOR *quarter_crd_to_uint_crd_cof = reinterpret_cast<const VECTOR *>(quarter_crd_to_uint_c… in refreshuintcrd()
51 const VECTOR *test_frc = reinterpret_cast<const VECTOR *>(test_frc_f); in refreshuintcrd()
Dconstrain_force_cycle_with_virial_impl.cu25 … const VECTOR *scaler, CONSTRAIN_PAIR *constrain_pair, in Constrain_Force_Cycle_With_Virial()
26 … const VECTOR *pair_dr, VECTOR *test_frc, float *d_atom_virial) { in Constrain_Force_Cycle_With_Virial()
30 VECTOR dr0 = pair_dr[pair_i]; in Constrain_Force_Cycle_With_Virial()
31 …VECTOR dr = Get_Periodic_Displacement(uint_crd[cp.atom_i_serial], uint_crd[cp.atom_j_serial], scal… in Constrain_Force_Cycle_With_Virial()
34 VECTOR frc_lin = frc_abs * dr0; in Constrain_Force_Cycle_With_Virial()
58 const VECTOR *scaler = reinterpret_cast<const VECTOR *>(scaler_f); in Constrain_Force_Cycle_With_Virial()
59 const VECTOR *pair_dr = reinterpret_cast<const VECTOR *>(pair_dr_f); in Constrain_Force_Cycle_With_Virial()
61 VECTOR *test_frc = reinterpret_cast<VECTOR *>(test_frc_f); in Constrain_Force_Cycle_With_Virial()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/nvtit/
Dmd_iteration_leap_frog_with_max_vel_impl.cu21 …teration_Leap_Frog_With_Max_Velocity(const int atom_numbers, VECTOR *vel, VECTOR *crd, VECTOR *frc, in MD_Iteration_Leap_Frog_With_Max_Velocity()
22 … VECTOR *acc, const float *inverse_mass, const float dt, in MD_Iteration_Leap_Frog_With_Max_Velocity()
26 VECTOR acc_i = inverse_mass[i] * frc[i]; in MD_Iteration_Leap_Frog_With_Max_Velocity()
27 VECTOR vel_i = vel[i] + dt * acc_i; in MD_Iteration_Leap_Frog_With_Max_Velocity()
38 VECTOR *d_vel = reinterpret_cast<VECTOR *>(vel); in MDIterationLeapFrogWithMaxVelocity()
39 VECTOR *d_crd = reinterpret_cast<VECTOR *>(crd); in MDIterationLeapFrogWithMaxVelocity()
40 VECTOR *d_frc = reinterpret_cast<VECTOR *>(frc); in MDIterationLeapFrogWithMaxVelocity()
41 VECTOR *d_acc = reinterpret_cast<VECTOR *>(acc); in MDIterationLeapFrogWithMaxVelocity()
Dmd_iteration_leap_frog_liujian_gpu_impl.cu23 … float *sqrt_mass_inverse, VECTOR *vel, VECTOR *crd, in MD_Iteration_Leap_Frog_With_LiuJian_kernel()
24 … VECTOR *frc, VECTOR *acc, VECTOR *random_frc, in MD_Iteration_Leap_Frog_With_LiuJian_kernel()
25 VECTOR *output) { in MD_Iteration_Leap_Frog_With_LiuJian_kernel()
58 VECTOR *d_vel = reinterpret_cast<VECTOR *>(vel); in MD_Iteration_Leap_Frog_With_LiuJian()
59 VECTOR *d_crd = reinterpret_cast<VECTOR *>(crd); in MD_Iteration_Leap_Frog_With_LiuJian()
60 VECTOR *d_frc = reinterpret_cast<VECTOR *>(frc); in MD_Iteration_Leap_Frog_With_LiuJian()
61 VECTOR *d_acc = reinterpret_cast<VECTOR *>(acc); in MD_Iteration_Leap_Frog_With_LiuJian()
62 VECTOR *d_rand_frc = reinterpret_cast<VECTOR *>(rand_frc); in MD_Iteration_Leap_Frog_With_LiuJian()
63 VECTOR *d_out = reinterpret_cast<VECTOR *>(output); in MD_Iteration_Leap_Frog_With_LiuJian()
Dmd_iteration_leap_frog_liujian_with_max_vel_impl.cu23 …const float *sqrt_mass_inverse, VECTOR *vel, VECTOR *crd, VECTOR *frc, VECTOR *acc, VECTOR *random… in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Velocity()
24 VECTOR *output, const float max_vel) { in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Velocity()
70 VECTOR *d_vel = reinterpret_cast<VECTOR *>(vel); in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
71 VECTOR *d_crd = reinterpret_cast<VECTOR *>(crd); in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
72 VECTOR *d_frc = reinterpret_cast<VECTOR *>(frc); in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
73 VECTOR *d_acc = reinterpret_cast<VECTOR *>(acc); in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
74 VECTOR *d_rand_frc = reinterpret_cast<VECTOR *>(rand_frc); in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
75 VECTOR *d_out = reinterpret_cast<VECTOR *>(output); in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
Dmd_iteration_leap_frog_impl.cu24 …_ void MD_Iteration_Leap_Frog(const int atom_numbers, VECTOR *vel, VECTOR *crd, VECTOR *frc, VECTO… in MD_Iteration_Leap_Frog()
48 VECTOR *d_vel = reinterpret_cast<VECTOR *>(vel); in MDIterationLeapFrog()
49 VECTOR *d_crd = reinterpret_cast<VECTOR *>(crd); in MDIterationLeapFrog()
50 VECTOR *d_frc = reinterpret_cast<VECTOR *>(frc); in MDIterationLeapFrog()
51 VECTOR *d_acc = reinterpret_cast<VECTOR *>(acc); in MDIterationLeapFrog()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/restrain/
Drestrain_force_atom_energy_virial_impl.cu21 … const VECTOR *crd, const VECTOR *crd_ref, const float weight, in restrain_force_with_atom_energy_and_virial()
22 … const VECTOR *boxlength, float *atom_energy, in restrain_force_with_atom_energy_and_virial()
23 float *atom_virial, VECTOR *frc) { in restrain_force_with_atom_energy_and_virial()
27 VECTOR dr = Get_Periodic_Displacement(crd_ref[atom_i], crd[atom_i], boxlength[0]); in restrain_force_with_atom_energy_and_virial()
29 VECTOR temp_force = 2 * weight * dr; in restrain_force_with_atom_energy_and_virial()
47 const VECTOR *crd = reinterpret_cast<const VECTOR *>(crd_f); in restrainforcewithatomenergyandvirial()
48 const VECTOR *crd_ref = reinterpret_cast<const VECTOR *>(crd_ref_f); in restrainforcewithatomenergyandvirial()
49 const VECTOR *boxlength = reinterpret_cast<const VECTOR *>(boxlength_f); in restrainforcewithatomenergyandvirial()
50 VECTOR *frc = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(frc_f)); in restrainforcewithatomenergyandvirial()
Drestrain_energy_impl.cu21 …oid restrain_energy_kernel(const int restrain_numbers, const int *restrain_list, const VECTOR *crd, in restrain_energy_kernel()
22 … const VECTOR *crd_ref, const float weight, const VECTOR boxlength, float *ene) { in restrain_energy_kernel()
26 VECTOR dr = Get_Periodic_Displacement(crd_ref[atom_i], crd[atom_i], boxlength); in restrain_energy_kernel()
36 const VECTOR *crd = reinterpret_cast<const VECTOR *>(crd_f); in restrainenergy()
37 const VECTOR *crd_ref = reinterpret_cast<const VECTOR *>(crd_ref_f); in restrainenergy()
38 const VECTOR *boxlength = reinterpret_cast<const VECTOR *>(boxlength_f); in restrainenergy()
Drestrain_force_impl.cu22 … const UNSIGNED_INT_VECTOR *uint_crd_ref, const float factor, const VECTOR *scaler, in restrainforcekernel()
23 VECTOR *frc) { in restrainforcekernel()
27 VECTOR dr = Get_Periodic_Displacement(uint_crd_ref[atom_i], uint_crd[atom_i], scaler[0]); in restrainforcekernel()
47 VECTOR *scaler = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(scaler_f)); in restrainforce()
48 VECTOR *frc = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(frc_f)); in restrainforce()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/
Dcommon_sponge.cuh45 struct VECTOR { struct
125 __device__ __host__ static inline VECTOR operator-(const VECTOR &veca, const VECTOR &vecb) { in operator -()
126 VECTOR vec; in operator -()
133 __device__ __host__ static inline VECTOR Get_Periodic_Displacement(const UNSIGNED_INT_VECTOR uvec_a, in Get_Periodic_Displacement()
135 const VECTOR scaler) { in Get_Periodic_Displacement()
136 VECTOR dr; in Get_Periodic_Displacement()
143 __device__ __host__ static inline VECTOR Get_Periodic_Displacement(const UINT_VECTOR_LJ_TYPE uvec_a, in Get_Periodic_Displacement()
145 const VECTOR scaler) { in Get_Periodic_Displacement()
146 VECTOR dr; in Get_Periodic_Displacement()
153 __device__ __host__ static inline VECTOR Get_Periodic_Displacement(const VECTOR vec_a, const VECTOR… in Get_Periodic_Displacement()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/dihedral/
Ddihedral_force_impl.cu21 …ihedralForceKernel(int dihedral_numbers, const UNSIGNED_INT_VECTOR *uint_crd, const VECTOR *scaler, in DihedralForceKernel()
24 const float *pn, VECTOR *frc) { in DihedralForceKernel()
38 VECTOR drij = Get_Periodic_Displacement(uint_crd[atom_i], uint_crd[atom_j], scaler[0]); in DihedralForceKernel()
39 VECTOR drkj = Get_Periodic_Displacement(uint_crd[atom_k], uint_crd[atom_j], scaler[0]); in DihedralForceKernel()
40 VECTOR drkl = Get_Periodic_Displacement(uint_crd[atom_k], uint_crd[atom_l], scaler[0]); in DihedralForceKernel()
42 VECTOR r1 = drij ^ drkj; in DihedralForceKernel()
43 VECTOR r2 = drkl ^ drkj; in DihedralForceKernel()
75 VECTOR dphi_dr1 = r1_1_r2_1 * r2 + cos_phi * r1_2 * r1; in DihedralForceKernel()
76 VECTOR dphi_dr2 = r1_1_r2_1 * r1 + cos_phi * r2_2 * r2; in DihedralForceKernel()
78 VECTOR dE_dri = dE_dphi * drkj ^ dphi_dr1; in DihedralForceKernel()
[all …]
Ddihedral_force_with_atom_energy_impl.cu22 … const VECTOR *scaler, const int *atom_a, const int *atom_b, in DihedralForceWithAtomEnergyKernel()
24 … const float *gamc, const float *gams, const float *pn, VECTOR *frc, in DihedralForceWithAtomEnergyKernel()
40 VECTOR drij = Get_Periodic_Displacement(uint_crd[atom_i], uint_crd[atom_j], scaler[0]); in DihedralForceWithAtomEnergyKernel()
41 VECTOR drkj = Get_Periodic_Displacement(uint_crd[atom_k], uint_crd[atom_j], scaler[0]); in DihedralForceWithAtomEnergyKernel()
42 VECTOR drkl = Get_Periodic_Displacement(uint_crd[atom_k], uint_crd[atom_l], scaler[0]); in DihedralForceWithAtomEnergyKernel()
44 VECTOR r1 = drij ^ drkj; in DihedralForceWithAtomEnergyKernel()
45 VECTOR r2 = drkl ^ drkj; in DihedralForceWithAtomEnergyKernel()
77 VECTOR dphi_dr1 = r1_1_r2_1 * r2 + cos_phi * r1_2 * r1; in DihedralForceWithAtomEnergyKernel()
78 VECTOR dphi_dr2 = r1_1_r2_1 * r1 + cos_phi * r2_2 * r2; in DihedralForceWithAtomEnergyKernel()
80 VECTOR dE_dri = dE_dphi * drkj ^ dphi_dr1; in DihedralForceWithAtomEnergyKernel()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/common/
Dmap_center_of_mass_impl.cu21 … float *scaler, VECTOR *center_of_mass, VECTOR *box_length, VECTOR *no_wrap_crd, VECTOR *crd) { in Map_Center_Of_Mass()
22 VECTOR trans_vec; in Map_Center_Of_Mass()
23 VECTOR com; in Map_Center_Of_Mass()
41 VECTOR *crd = reinterpret_cast<VECTOR *>(crd_f); in MapCenterOfMass()
42 VECTOR *no_wrap_crd = reinterpret_cast<VECTOR *>(no_wrap_crd_f); in MapCenterOfMass()
43 VECTOR *box_length = reinterpret_cast<VECTOR *>(box_length_f); in MapCenterOfMass()
44 VECTOR *center_of_mass = reinterpret_cast<VECTOR *>(center_of_mass_f); in MapCenterOfMass()
Dgetcenter_impl.cu21 … const int *center_atoms, const VECTOR *crd, VECTOR *center_of_geometry) { in GetCenterOfGeometryKernel()
25 VECTOR temp = center_numbers_inverse * crd[atom_i]; in GetCenterOfGeometryKernel()
34 VECTOR *crd = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(crd_f)); in GetCenterOfGeometry()
35 …VECTOR *center_of_geometry = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(center_of_geome… in GetCenterOfGeometry()
Dcrd_to_uint_crd_quarter_impl.cu19 …void Crd_To_Uint_Crd_Quarter(const int atom_numbers, const VECTOR *scale_factor, const VECTOR *crd, in Crd_To_Uint_Crd_Quarter()
24 VECTOR temp = crd[atom_i]; in Crd_To_Uint_Crd_Quarter()
41 VECTOR *crd = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(crd_f)); in CrdToUintCrdQuarter()
42 …VECTOR *crd_to_uint_crd_cof = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(crd_to_uint_cr… in CrdToUintCrdQuarter()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/crdmcmap/
Drefresh_boxmaptimes_impl.cu21 __global__ void Refresh_BoxMapTimes_CUDA(int atom_numbers, VECTOR *box_length_inverse, VECTOR *crd, in Refresh_BoxMapTimes_CUDA()
22 INT_VECTOR *box_map_times, VECTOR *old_crd) { in Refresh_BoxMapTimes_CUDA()
23 VECTOR crd_i, old_crd_i; in Refresh_BoxMapTimes_CUDA()
37 VECTOR *box_length_inverse = reinterpret_cast<VECTOR *>(box_length_inverse_f); in refresh_boxmaptimes()
38 VECTOR *crd = reinterpret_cast<VECTOR *>(crd_f); in refresh_boxmaptimes()
39 VECTOR *old_crd = reinterpret_cast<VECTOR *>(old_crd_f); in refresh_boxmaptimes()
Dcal_no_wrap_crd_impl.cu22 …_ void Calculate_No_Wrap_Crd(int atom_numbers, INT_VECTOR *box_map_times, VECTOR *box, VECTOR *crd, in Calculate_No_Wrap_Crd()
23 VECTOR *nowrap_crd) { in Calculate_No_Wrap_Crd()
36 VECTOR *box = reinterpret_cast<VECTOR *>(box_f); in calculatenowrapcrd()
37 VECTOR *crd = reinterpret_cast<VECTOR *>(crd_f); in calculatenowrapcrd()
38 VECTOR *nowrap_crd = reinterpret_cast<VECTOR *>(nowrap_crd_f); in calculatenowrapcrd()
/third_party/protobuf/java/core/src/main/java/com/google/protobuf/
DFieldType.java60 DOUBLE_LIST(18, Collection.VECTOR, JavaType.DOUBLE),
61 FLOAT_LIST(19, Collection.VECTOR, JavaType.FLOAT),
62 INT64_LIST(20, Collection.VECTOR, JavaType.LONG),
63 UINT64_LIST(21, Collection.VECTOR, JavaType.LONG),
64 INT32_LIST(22, Collection.VECTOR, JavaType.INT),
65 FIXED64_LIST(23, Collection.VECTOR, JavaType.LONG),
66 FIXED32_LIST(24, Collection.VECTOR, JavaType.INT),
67 BOOL_LIST(25, Collection.VECTOR, JavaType.BOOLEAN),
68 STRING_LIST(26, Collection.VECTOR, JavaType.STRING),
69 MESSAGE_LIST(27, Collection.VECTOR, JavaType.MESSAGE),
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/angle/
Dangle_force_impl.cu21 …void AngleForceKernel(int angle_numbers, const UNSIGNED_INT_VECTOR *uint_crd, const VECTOR *scaler, in AngleForceKernel()
23 const float *angle_theta0, VECTOR *frc) { in AngleForceKernel()
33 VECTOR drij = Get_Periodic_Displacement(uint_crd[atom_i], uint_crd[atom_j], scaler[0]); in AngleForceKernel()
34 VECTOR drkj = Get_Periodic_Displacement(uint_crd[atom_k], uint_crd[atom_j], scaler[0]); in AngleForceKernel()
50 VECTOR fi = common_factor_self * rij_2 * drij - common_factor_cross * drkj; in AngleForceKernel()
51 VECTOR fk = common_factor_self * rkj_2 * drkj - common_factor_cross * drij; in AngleForceKernel()
77 VECTOR *frc = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(frc_f)); in AngleForce()
78 VECTOR *scaler = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(scaler_f)); in AngleForce()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/neighbor_list/
Dneighbor_list_impl.cu34 static __global__ void Crd_To_Uint_Crd(const int atom_numbers, float *scale_factor, const VECTOR *c… in Crd_To_Uint_Crd()
39 VECTOR temp = crd[atom_i]; in Crd_To_Uint_Crd()
55 static __global__ void Crd_Periodic_Map(const int atom_numbers, VECTOR *crd, const float *box_lengt… in Crd_Periodic_Map()
99 … const VECTOR *crd, const int *grid_N, const int gridxy, in Find_Atom_In_Grid_Serial()
153 VECTOR dr; in Find_atom_neighbors()
258 __global__ void Crd_To_Uint_Crd_Half(const int atom_numbers, float *scale_factor, const VECTOR *crd, in Crd_To_Uint_Crd_Half()
271 __global__ void Vector_Translation(const int vector_numbers, VECTOR *vec_list, const VECTOR transla… in Vector_Translation()
280 __global__ void Vector_Translation(const int vector_numbers, VECTOR *vec_list, const VECTOR *transl… in Vector_Translation()
289 …s_need_refresh_neighbor_list_cuda(const int atom_numbers, const VECTOR *crd, const VECTOR *old_crd, in Is_need_refresh_neighbor_list_cuda()
293 VECTOR r1 = crd[i]; in Is_need_refresh_neighbor_list_cuda()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/bond/
Dbond_force_cuda_gpu_impl.cu21 …id BondForceCudaKernel(int bond_numbers, const UNSIGNED_INT_VECTOR *uint_crd, const VECTOR *scaler, in BondForceCudaKernel()
23 VECTOR *frc) { in BondForceCudaKernel()
31 VECTOR dr = Get_Periodic_Displacement(uint_crd[atom_i], uint_crd[atom_j], scaler[0]); in BondForceCudaKernel()
35 VECTOR f = 2 * tempf * k * dr; in BondForceCudaKernel()
54 VECTOR *scaler = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(scaler_f)); in BondForce()
55 VECTOR *frc = const_cast<VECTOR *>(reinterpret_cast<const VECTOR *>(frc_f)); in BondForce()

12345