diff --git a/lmcache/experimental/storage_backend/connector/nixl_connector_v2.py b/lmcache/experimental/storage_backend/connector/nixl_connector_v2.py index 498399c07c..7ec8bb2a80 100644 --- a/lmcache/experimental/storage_backend/connector/nixl_connector_v2.py +++ b/lmcache/experimental/storage_backend/connector/nixl_connector_v2.py @@ -275,7 +275,8 @@ def _commit_write(self, write_size: int, uid: str): self._local_xfer_handlers, desc_indexes, self._remote_xfer_handlers, - desc_indexes) + desc_indexes, + notif_msg=uuid_to_message(uid)) t2 = time.perf_counter() self._agent.transfer(handle) #, uuid_to_message(uid)) @@ -294,8 +295,6 @@ def _commit_write(self, write_size: int, uid: str): f"status: {status}") t3 = time.perf_counter() - self._agent.send_notif(self.peer_name, uuid_to_message(uid)) - logger.debug( "Transfer %s completed in %.4f ms, creating the transfer: %.4f ms," " transfer time: %.4f ms, pure transfer throughput: %.4f GB/s",