BF: Fix memory allocation error in probtrackx2_gpu

Moises Fernandezrequested to merge
fix_largemem_Ampere into master

For some new GPUs, memory is increased up to 40GB, and some of the int variables that store the number of threads overflow. Here I change these to size_t type to avoid overflow.

Edited by Paul McCarthy

Merge request reports