test: add decorator to get_torch_version for testing

Signed-off-by: Pablo Garay <[email protected]>

Author: pablo-garay

megatron/core/utils.py

@@ -2,6 +2,8 @@
 
 """Utility functions used throughout Megatron core"""
 
+from .backwards_compatibility_decorators import exempt_from_compat_check
+
 import array
 import asyncio
 import functools
@@ -280,6 +282,7 @@ def __getattribute__(self, attr):
     return validator
 
 
+@exempt_from_compat_check  # TEST: Decorator functionality
 def get_torch_version():
     """Get pytorch version from __version__; if not available use pip's. Use caching."""