Nathalie Furmento лет назад: 6
Родитель
Сommit
9214e8b8fb
2 измененных файлов с 5 добавлено и 5 удалено
  1. 4 4
      src/drivers/mpi/driver_mpi_common.c
  2. 1 1
      src/drivers/mpi/driver_mpi_source.c

+ 4 - 4
src/drivers/mpi/driver_mpi_common.c

@@ -174,7 +174,7 @@ void _starpu_mpi_common_send(const struct _starpu_mp_node *node, void *msg, int
                 channel->event.mpi_ms_event.is_sender = 1;
 
                 /* call by sink, we need to initialize some parts, for host it's done in data_request.c */
-                if (channel->type == STARPU_UNUSED)
+                if (channel->node_ops == NULL)
                         channel->event.mpi_ms_event.requests = NULL;
 
                 /* Initialize the list */
@@ -221,7 +221,7 @@ void _starpu_mpi_common_recv(const struct _starpu_mp_node *node, void *msg, int
                 channel->event.mpi_ms_event.is_sender = 0;
 
                 /* call by sink, we need to initialize some parts, for host it's done in data_request.c */
-                if (channel->type == STARPU_UNUSED)
+                if (channel->node_ops == NULL)
                         channel->event.mpi_ms_event.requests = NULL;
 
                 /* Initialize the list */
@@ -271,7 +271,7 @@ void _starpu_mpi_common_send_to_device(const struct _starpu_mp_node *node STARPU
                 channel->event.mpi_ms_event.is_sender = 1;
 
                 /* call by sink, we need to initialize some parts, for host it's done in data_request.c */
-                if (channel->type == STARPU_UNUSED)
+                if (channel->node_ops == NULL)
                         channel->event.mpi_ms_event.requests = NULL;
 
                 /* Initialize the list */
@@ -313,7 +313,7 @@ void _starpu_mpi_common_recv_from_device(const struct _starpu_mp_node *node STAR
                 channel->event.mpi_ms_event.is_sender = 0;
 
                 /* call by sink, we need to initialize some parts, for host it's done in data_request.c */
-                if (channel->type == STARPU_UNUSED)
+                if (channel->node_ops == NULL)
                         channel->event.mpi_ms_event.requests = NULL;
 
                 /* Initialize the list */

+ 1 - 1
src/drivers/mpi/driver_mpi_source.c

@@ -547,7 +547,7 @@ void _starpu_mpi_free_on_node(unsigned dst_node, uintptr_t addr, size_t size, in
 	_starpu_mpi_source_free_memory((void*) addr, dst_node);
 }
 
-struct _starpu_node_ops _starpu_driver_mic_node_ops =
+struct _starpu_node_ops _starpu_driver_mpi_node_ops =
 {
 	.copy_data_to[STARPU_UNUSED] = NULL,
 	.copy_data_to[STARPU_CPU_RAM] = _starpu_mpi_copy_data_from_mpi_to_cpu,