pytorch

059c55f8 - Enable test_jit tests that work on ROCm 2.1

Enable test_jit tests that work on ROCm 2.1 Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17010 Differential Revision: D14059748 Pulled By: bddppq fbshipit-source-id: 7a1f7eee4f818dba91e741437415370973e4d429