Allow row sizes to support int64/size_t. (#69303)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69303
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/792
Follow up to D32715453 (https://github.com/pytorch/pytorch/commit/e60fd1065926f7f25173dc08f3b2f2091d0af9c2), allowing row size to be 64-bit.
Test Plan:
buck test mode/opt -c fbcode.caffe2_gpu_type=v100,a100 //deeplearning/fbgemm/fbgemm_gpu:quantize_ops_test
buck test mode/opt -c fbcode.caffe2_gpu_type=none //deeplearning/fbgemm/fbgemm_gpu:quantize_ops_test
buck test mode/opt //caffe2/test:
Reviewed By: jspark1105, jianyuh
Differential Revision: D32768838
fbshipit-source-id: 9e2b01d8d23e71f8333820e725379c3fc1c0711a
