Changed size_t to int to reflect type of dimension variable

I had to make this small change in order for the gpu part to run on my setup. AMD system/Linux environment.