using torchTRT enum platform

Author: apbose

py/torch_tensorrt/_utils.py

@@ -1,9 +1,7 @@
-import ctypes
-import platform
-import sys
 from typing import Any
 
 import torch
+from torch_tensorrt._enums import Platform
 
 
 def sanitized_torch_version() -> Any:
@@ -14,12 +12,14 @@ def sanitized_torch_version() -> Any:
     )
 
 
-def check_cross_compile_trt_win_lib():
+def check_cross_compile_trt_win_lib() -> bool:
     # cross compile feature is only available on linux
     # build engine on linux and run on windows
     import dllist
 
-    if sys.platform.startswith("linux"):
+    platform = Platform.current_platform()
+    platform = str(platform).lower()
+    if platform.startswith("linux"):
         loaded_libs = dllist.dllist()
         target_lib = "libnvinfer_builder_resource_win.so.*"
         if target_lib in loaded_libs: