DispatchKeySet perf improvements (#72403)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72403
reland of D33301590 (https://github.com/pytorch/pytorch/commit/18cbe80f23c6c236361c45b5debff58b65359a23)
ghstack-source-id: 148830729
Test Plan: CI, and running explicit mobile test: `buck test //fbandroid/instrumentation_tests/com/facebook/pytorch/bi_xray:instrumentation_test -c test.external_runner=tpx -- --regex 'testBIXRayModel.*PyTorchBIXRayInstrumentationTest' --force-remote-execution --run-disabled`
Reviewed By: albanD
Differential Revision: D34034847
fbshipit-source-id: a930e44513a76c0c82c9d27f0fc2d2a6d7d90cf9
(cherry picked from commit 7f1ea7584c4fb098645428070a94b2f231af787b)
