[Resubmit] Add compare_set operation and test to TCPStore (#51815)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51815
This is resubmission of #51593, already approved.
Test Plan: Imported from OSS
Reviewed By: izdeby
Differential Revision: D26316875
Pulled By: H-Huang
fbshipit-source-id: d81cb131ef6b9e2ebaee32bb505dfc11235bc29d
