Fix (export/onnx): import GLOBALS from correct location depending on torch version (#1398)

Author: JP-Amboage

src/brevitas/export/onnx/__init__.py

@@ -1,5 +1,8 @@
 # Copyright (C) 2023, Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: BSD-3-Clause
+from packaging import version
+
+from brevitas import torch_version
 
 
 def onnx_export_opset():
@@ -9,7 +12,11 @@ def onnx_export_opset():
         opset = getattr(cfg, ATR_NAME)
 
     except:
-        from torch.onnx._globals import GLOBALS as cfg
+        if torch_version < version.parse('2.9.0'):
+            from torch.onnx._globals import GLOBALS as cfg
+        else:
+            from torch.onnx._internal.torchscript_exporter._globals import GLOBALS as cfg
+
         ATR_NAME = 'export_onnx_opset_version'
         opset = getattr(cfg, ATR_NAME)