Add back SHARD_NUMBER and TEST_CONFIG to upload test stats step
These are used in the print_test_stats file and somehow are no longer passed.
Test plan:
After CI runs, check the S3 to make sure the file names include shard number and config.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75989
Approved by: https://github.com/clee2000
