From d6ae8065f7c422300dba3b6256f4ace53b39a092 Mon Sep 17 00:00:00 2001 From: Thomas White Date: Thu, 18 Nov 2010 11:04:05 +0100 Subject: Handle multiple GPUs --- src/pattern_sim.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src/pattern_sim.c') diff --git a/src/pattern_sim.c b/src/pattern_sim.c index 3c1afe07..3deaa742 100644 --- a/src/pattern_sim.c +++ b/src/pattern_sim.c @@ -203,6 +203,7 @@ int main(int argc, char *argv[]) int done = 0; UnitCell *input_cell; struct quaternion orientation; + int gpu_dev = -1; /* Long options */ const struct option longopts[] = { @@ -223,6 +224,7 @@ int main(int argc, char *argv[]) {"geometry", 1, NULL, 'g'}, {"beam", 1, NULL, 'b'}, {"really-random", 0, &config_random, 1}, + {"gpu-dev", 1, NULL, 2}, {0, 0, NULL, 0} }; @@ -271,6 +273,10 @@ int main(int argc, char *argv[]) beamfile = strdup(optarg); break; + case 2 : + gpu_dev = atoi(optarg); + break; + case 0 : break; @@ -446,7 +452,7 @@ int main(int argc, char *argv[]) if ( config_gpu ) { if ( gctx == NULL ) { gctx = setup_gpu(config_nosfac, &image, - intensities); + intensities, gpu_dev); } get_diffraction_gpu(gctx, &image, na, nb, nc, cell); } else { -- cgit v1.2.3