pytorch

901aa85b - fix TEST_ROCM definition to disable test_jit_cudnn_extension on rocm (#110385)

fix TEST_ROCM definition to disable test_jit_cudnn_extension on rocm (#110385) Define TEST_ROCM before modification TEST_CUDA. Otherwise TEST_ROCM will always be false and will not disable test_jit_cudnn_extension for rocm Fixes https://github.com/pytorch/pytorch/issues/107182 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110385 Approved by: https://github.com/jithunnair-amd, https://github.com/kit1980