Allow the usage of custom GPUs for the benchmark.

[vcarpani]

Hello, I created this quick PR to benchmark the new algorithm also on other NVIDIA GPUs. The default behaviour will still be the same of the current script, but more options are available for custom benchmarking.