Add support for user defined sets (#153553)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/153553
Approved by: https://github.com/williamwen42, https://github.com/zou3519
ghstack dependencies: #153150, #152991, #154539
Reviewed By: jeanschmidt
Differential Revision: D77790831
fbshipit-source-id: 4ba168e1be0203adf73e618a2cece725580cda63
Author
generatedunixname499836121
