Searched refs:shared_grad (Results 1 – 2 of 2) sorted by relevance
959 shared_grad[tid][0] += shared_grad[tid + s][0]; in two_body_ratio_grad_PBC_kernel()960 shared_grad[tid][1] += shared_grad[tid + s][1]; in two_body_ratio_grad_PBC_kernel()961 shared_grad[tid][2] += shared_grad[tid + s][2]; in two_body_ratio_grad_PBC_kernel()962 shared_grad[tid][3] += shared_grad[tid + s][3]; in two_body_ratio_grad_PBC_kernel()1037 shared_grad[tid][0] = shared_grad[tid][1] = shared_grad[tid][2] = 0.0f; in two_body_ratio_grad_PBC_kernel_fast()1077 shared_grad[tid][0] += shared_grad[tid + s][0]; in two_body_ratio_grad_PBC_kernel_fast()1078 shared_grad[tid][1] += shared_grad[tid + s][1]; in two_body_ratio_grad_PBC_kernel_fast()1079 shared_grad[tid][2] += shared_grad[tid + s][2]; in two_body_ratio_grad_PBC_kernel_fast()2805 shared_grad[tid][0] = shared_grad[tid][1] = shared_grad[tid][2] = shared_grad[tid][3] = (T)0.0; in one_body_ratio_grad_PBC_kernel()2844 shared_grad[tid][0] += shared_grad[tid + s][0]; in one_body_ratio_grad_PBC_kernel()[all …]
417 __shared__ T shared_grad[BS][3]; in two_body_ratio_grad_kernel() local419 shared_grad[tid][0] = shared_grad[tid][1] = shared_grad[tid][2] = 0.0f; in two_body_ratio_grad_kernel()447 shared_grad[tid][0] += du * dx; in two_body_ratio_grad_kernel()459 shared_grad[tid][0] += shared_grad[tid + s][0]; in two_body_ratio_grad_kernel()460 shared_grad[tid][1] += shared_grad[tid + s][1]; in two_body_ratio_grad_kernel()461 shared_grad[tid][2] += shared_grad[tid + s][2]; in two_body_ratio_grad_kernel()1521 __shared__ T shared_grad[BS][3]; in one_body_ratio_grad_kernel() local1523 shared_grad[tid][0] = shared_grad[tid][1] = shared_grad[tid][2] = 0.0f; in one_body_ratio_grad_kernel()1562 shared_grad[tid][0] += shared_grad[tid + s][0]; in one_body_ratio_grad_kernel()1563 shared_grad[tid][1] += shared_grad[tid + s][1]; in one_body_ratio_grad_kernel()[all …]