Add API usage logging to ShardedTensor and fix a few tests. (#68771)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68771
ghstack-source-id: 143974518
Test Plan: waitforbuildbot
Reviewed By: fduwjj, wanchaol
Differential Revision: D32601562
fbshipit-source-id: ed624137efab94fbe556609bb40cca14e69d9bac