Selaa lähdekoodia

MPI MS: Fix looking up symbols from libraries

Samuel Thibault 4 vuotta sitten
vanhempi
commit
7948043abf
2 muutettua tiedostoa jossa 8 lisäystä ja 0 poistoa
  1. 4 0
      src/drivers/mic/driver_mic_sink.c
  2. 4 0
      src/drivers/mpi/driver_mpi_sink.c

+ 4 - 0
src/drivers/mic/driver_mic_sink.c

@@ -223,6 +223,10 @@ void _starpu_mic_sink_bind_thread(const struct _starpu_mp_node *mp_node STARPU_A
 
 
 void (*_starpu_mic_sink_lookup (const struct _starpu_mp_node * node STARPU_ATTRIBUTE_UNUSED, char* func_name))(void)
 void (*_starpu_mic_sink_lookup (const struct _starpu_mp_node * node STARPU_ATTRIBUTE_UNUSED, char* func_name))(void)
 {
 {
+#ifdef RTLD_DEFAULT
+	void *dl_handle = dlopen(RTLD_DEFAULT, RTLD_NOW);
+#else
 	void *dl_handle = dlopen(NULL, RTLD_NOW);
 	void *dl_handle = dlopen(NULL, RTLD_NOW);
+#endif
 	return dlsym(dl_handle, func_name);
 	return dlsym(dl_handle, func_name);
 }
 }

+ 4 - 0
src/drivers/mpi/driver_mpi_sink.c

@@ -44,7 +44,11 @@ void _starpu_mpi_sink_deinit(struct _starpu_mp_node *node)
 
 
 void (*_starpu_mpi_sink_lookup (const struct _starpu_mp_node * node STARPU_ATTRIBUTE_UNUSED, char* func_name))(void)
 void (*_starpu_mpi_sink_lookup (const struct _starpu_mp_node * node STARPU_ATTRIBUTE_UNUSED, char* func_name))(void)
 {
 {
+#ifdef RTLD_DEFAULT
+        void *dl_handle = dlopen(RTLD_DEFAULT, RTLD_NOW);
+#else
         void *dl_handle = dlopen(NULL, RTLD_NOW);
         void *dl_handle = dlopen(NULL, RTLD_NOW);
+#endif
         return dlsym(dl_handle, func_name);
         return dlsym(dl_handle, func_name);
 }
 }