prov/psm3: Clean ref's to split cuda hostbufs when no longer needed

Signed-off-by: Goldman, Adam <[email protected]>
intel · Jul 27, 2021 · 98f2841 · 98f2841
1 parent 856f8b2
commit 98f2841
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/psm3/ptl_ips/ips_proto_expected.c b/psm3/ptl_ips/ips_proto_expected.c
@@ -1882,6 +1882,10 @@ psm2_error_t ips_tid_send_exp(struct ips_tid_send_desc *tidsendc)
 			}
 			psmi_cuda_run_prefetcher(protoexp, tidsendc);
 		}
+		/* Clean Up tidsendc ref's to split cuda hostbufs when no longer needed */
+		tidsendc->cuda_num_buf = 0;
+		tidsendc->cuda_hostbuf[0] = NULL;
+		tidsendc->cuda_hostbuf[1] = NULL;
 	}
 #endif
 	err = ips_tid_issue_rdma_write(tidsendc);