We probably should set the complete_ flag in the send/recv work object and expose that as a Python API. Similar issue exists in our Future API.
@Yanli_Zhao mentioned that this is also asked in https://github.com/pytorch/pytorch/issues/30723 1