Commit 2a1b192b authored by incardon's avatar incardon

Give the possibility to set-up device for external

parent 09c23abd
......@@ -290,7 +290,14 @@ public:
bar_req = MPI_Request();
bar_stat = MPI_Status();
context = new mgpu::ofp_context_t(mgpu::gpu_context_opt::no_print_props,shmrank);
#ifdef EXTERNAL_SET_GPU
int dev;
cudaGetDevice(&dev);
context = new mgpu::ofp_context_t(mgpu::gpu_context_opt::no_print_props,dev);
#else
context = new mgpu::ofp_context_t(mgpu::gpu_context_opt::no_print_props,shmrank);
#endif
#if defined(PRINT_RANK_TO_GPU) && defined(CUDA_GPU)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment