pytorch

4c9eb579 - [PyTorch] Narrow Device to 2 bytes by narrowing DeviceType and DeviceIndex (#47023)

[PyTorch] Narrow Device to 2 bytes by narrowing DeviceType and DeviceIndex (#47023) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47023 DeviceType pretty clearly only needs 1 byte. DeviceIndex only needs 1 byte given that machines don't have anywhere near 255 GPUs in them as far as I know. ghstack-source-id: 116901430 Test Plan: Existing tests, added assertion to catch if my assumption about DeviceIndex is incorrect Reviewed By: dzhulgakov Differential Revision: D24605460 fbshipit-source-id: 7c9a89027fcf8eebd623b7cdbf6302162c981cd2