pytorch

73eaa0a5 - Fixing error in jit cuda on ROCm: non-constant-expression cannot be n… (#55243)

Fixing error in jit cuda on ROCm: non-constant-expression cannot be n… (#55243) Summary: On ROCm, the error when compiling was "non-constant-expression cannot be narrowed from type 'int' to 'uint32_t'" when compiling grid_reduction.cu. Added typecast to fix issue. Also, removed test skip with ROCm : re-enabling Pull Request resolved: https://github.com/pytorch/pytorch/pull/55243 Reviewed By: malfet Differential Revision: D27917066 Pulled By: ngimel fbshipit-source-id: b0b7c5fc8ecd2624222b35fe060846f7d1670f07