using torchTRT enum platform

Author: apbose

py/torch_tensorrt/_compile.py

@@ -526,7 +526,6 @@ def convert_method_to_trt_engine(
         raise RuntimeError("Module is an unknown format or the ir requested is unknown")
 
 
-@needs_cross_compile
 def load_cross_compiled_exported_program(file_path: str = "") -> Any:
     """
     Load an ExportedProgram file in Windows which was previously cross compiled in Linux

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:

py/torch_tensorrt/dynamo/_compiler.py

@@ -1214,7 +1214,6 @@ def save_cross_compiled_exported_program(
     logger.debug(f"successfully saved the module for windows at {file_path}")
 
 
-@needs_cross_compile
 def load_cross_compiled_exported_program(file_path: str = "") -> Any:
     """
     Load an ExportedProgram file in Windows which was previously cross compiled in Linux