/***** The kernel declarations/definitions come here *****/
+#include "gromacs/mdlib/nbnxn_cuda/nbnxn_cuda_kernel_utils.cuh"
/* Top-level kernel declaration generation: will generate through multiple
* inclusion the following flavors for all kernel declarations:
* - The 1D block-grid contains as many blocks as super-clusters.
*/
int num_threads_z = 1;
- if (nb->dev_info->prop.major == 3 && nb->dev_info->prop.minor == 7)
+ if ((nb->dev_info->prop.major == 3 && nb->dev_info->prop.minor == 7) ||
+ (nb->dev_info->prop.major == 6 && nb->dev_info->prop.minor == 0))
{
num_threads_z = 2;
}