Give the possibility to set-up device for external

2a1b192b · incardon · 09c23abd · 2a1b192b
Commit 2a1b192b authored May 15, 2019 by incardon
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

src/VCluster/VCluster_base.hpp src/VCluster/VCluster_base.hpp +8 -1

No files found.
--- a/src/VCluster/VCluster_base.hpp
+++ b/src/VCluster/VCluster_base.hpp
@@ -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)