|
29 | 29 |
|
30 | 30 | #include "opal/mca/mpool/mpool.h" |
31 | 31 | #include "opal/util/arch.h" |
32 | | -#include "ompi/runtime/ompi_software_events.h" |
| 32 | +#include "ompi/runtime/ompi_spc.h" |
33 | 33 | #include "ompi/mca/pml/pml.h" |
34 | 34 | #include "ompi/mca/bml/bml.h" |
35 | 35 | #include "pml_ob1_comm.h" |
@@ -199,10 +199,9 @@ static void mca_pml_ob1_put_completion (mca_pml_ob1_rdma_frag_t *frag, int64_t r |
199 | 199 | if (OPAL_LIKELY(0 < rdma_size)) { |
200 | 200 |
|
201 | 201 | /* check completion status */ |
202 | | - |
203 | 202 | OPAL_THREAD_ADD_FETCH_SIZE_T(&recvreq->req_bytes_received, rdma_size); |
204 | | - SW_EVENT_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)rdma_size, OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
205 | | - |
| 203 | + SPC_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)rdma_size, |
| 204 | + OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
206 | 205 | if (recv_request_pml_complete_check(recvreq) == false && |
207 | 206 | recvreq->req_rdma_offset < recvreq->req_send_offset) { |
208 | 207 | /* schedule additional rdma operations */ |
@@ -246,8 +245,7 @@ int mca_pml_ob1_recv_request_ack_send_btl( |
246 | 245 | des->des_cbfunc = mca_pml_ob1_recv_ctl_completion; |
247 | 246 |
|
248 | 247 | rc = mca_bml_base_send(bml_btl, des, MCA_PML_OB1_HDR_TYPE_ACK); |
249 | | - SW_EVENT_RECORD(OMPI_BYTES_RECEIVED_MPI, (long long)size); |
250 | | - |
| 248 | + SPC_RECORD(OMPI_BYTES_RECEIVED_MPI, (long long)size); |
251 | 249 | if( OPAL_LIKELY( rc >= 0 ) ) { |
252 | 250 | return OMPI_SUCCESS; |
253 | 251 | } |
@@ -380,7 +378,8 @@ static void mca_pml_ob1_rget_completion (mca_btl_base_module_t* btl, struct mca_ |
380 | 378 | } else { |
381 | 379 | /* is receive request complete */ |
382 | 380 | OPAL_THREAD_ADD_FETCH_SIZE_T(&recvreq->req_bytes_received, frag->rdma_length); |
383 | | - SW_EVENT_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)frag->rdma_length, OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
| 381 | + SPC_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)frag->rdma_length, |
| 382 | + OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
384 | 383 | /* TODO: re-add order */ |
385 | 384 | mca_pml_ob1_send_fin (recvreq->req_recv.req_base.req_proc, |
386 | 385 | bml_btl, frag->rdma_hdr.hdr_rget.hdr_frag, |
@@ -436,10 +435,8 @@ static int mca_pml_ob1_recv_request_put_frag (mca_pml_ob1_rdma_frag_t *frag) |
436 | 435 |
|
437 | 436 | /* send rdma request to peer */ |
438 | 437 | rc = mca_bml_base_send (bml_btl, ctl, MCA_PML_OB1_HDR_TYPE_PUT); |
439 | | - |
440 | 438 | /* Increment counter for bytes_put even though they probably haven't all been received yet */ |
441 | | - SW_EVENT_RECORD(OMPI_BYTES_PUT, frag->rdma_length); |
442 | | - |
| 439 | + SPC_RECORD(OMPI_BYTES_PUT, frag->rdma_length); |
443 | 440 | if (OPAL_UNLIKELY(rc < 0)) { |
444 | 441 | mca_bml_base_free (bml_btl, ctl); |
445 | 442 | return rc; |
@@ -481,10 +478,8 @@ int mca_pml_ob1_recv_request_get_frag (mca_pml_ob1_rdma_frag_t *frag) |
481 | 478 | rc = mca_bml_base_get (bml_btl, frag->local_address, frag->remote_address, local_handle, |
482 | 479 | (mca_btl_base_registration_handle_t *) frag->remote_handle, frag->rdma_length, |
483 | 480 | 0, MCA_BTL_NO_ORDER, mca_pml_ob1_rget_completion, frag); |
484 | | - |
485 | 481 | /* Increment counter for bytes_get even though they probably haven't all been received yet */ |
486 | | - SW_EVENT_RECORD(OMPI_BYTES_GET, frag->rdma_length); |
487 | | - |
| 482 | + SPC_RECORD(OMPI_BYTES_GET, frag->rdma_length); |
488 | 483 | if( OPAL_UNLIKELY(OMPI_SUCCESS > rc) ) { |
489 | 484 | return mca_pml_ob1_recv_request_get_frag_failed (frag, OMPI_ERR_OUT_OF_RESOURCE); |
490 | 485 | } |
@@ -540,8 +535,8 @@ void mca_pml_ob1_recv_request_progress_frag( mca_pml_ob1_recv_request_t* recvreq |
540 | 535 | ); |
541 | 536 |
|
542 | 537 | OPAL_THREAD_ADD_FETCH_SIZE_T(&recvreq->req_bytes_received, bytes_received); |
543 | | - SW_EVENT_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
544 | | - |
| 538 | + SPC_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, |
| 539 | + OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
545 | 540 | /* check completion status */ |
546 | 541 | if(recv_request_pml_complete_check(recvreq) == false && |
547 | 542 | recvreq->req_rdma_offset < recvreq->req_send_offset) { |
@@ -619,8 +614,8 @@ void mca_pml_ob1_recv_request_frag_copy_finished( mca_btl_base_module_t* btl, |
619 | 614 | des->des_cbfunc(NULL, NULL, des, 0); |
620 | 615 |
|
621 | 616 | OPAL_THREAD_ADD_FETCH_SIZE_T(&recvreq->req_bytes_received, bytes_received); |
622 | | - SW_EVENT_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
623 | | - |
| 617 | + SPC_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, |
| 618 | + OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
624 | 619 | /* check completion status */ |
625 | 620 | if(recv_request_pml_complete_check(recvreq) == false && |
626 | 621 | recvreq->req_rdma_offset < recvreq->req_send_offset) { |
@@ -834,7 +829,8 @@ void mca_pml_ob1_recv_request_progress_rndv( mca_pml_ob1_recv_request_t* recvreq |
834 | 829 | recvreq->req_recv.req_base.req_datatype); |
835 | 830 | ); |
836 | 831 | OPAL_THREAD_ADD_FETCH_SIZE_T(&recvreq->req_bytes_received, bytes_received); |
837 | | - SW_EVENT_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
| 832 | + SPC_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, |
| 833 | + OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
838 | 834 | } |
839 | 835 | /* check completion status */ |
840 | 836 | if(recv_request_pml_complete_check(recvreq) == false && |
@@ -905,9 +901,8 @@ void mca_pml_ob1_recv_request_progress_match( mca_pml_ob1_recv_request_t* recvre |
905 | 901 | * for this request. |
906 | 902 | */ |
907 | 903 | recvreq->req_bytes_received += bytes_received; |
908 | | - |
909 | | - SW_EVENT_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
910 | | - |
| 904 | + SPC_USER_OR_MPI(recvreq->req_recv.req_base.req_tag, (long long)bytes_received, |
| 905 | + OMPI_BYTES_RECEIVED_USER, OMPI_BYTES_RECEIVED_MPI); |
911 | 906 | recv_request_pml_complete(recvreq); |
912 | 907 | } |
913 | 908 |
|
|
0 commit comments