dll file path update for windows

Author: apbose

py/torch_tensorrt/dynamo/utils.py

@@ -847,8 +847,13 @@ def download_plugin_lib_path(py_version: str, platform: str) -> str:
 
     # Proceeding with the unzip of the wheel file
     # This will exist if the filename was already downloaded
-    if os.path.exists("./tensorrt_llm/libs/libnvinfer_plugin_tensorrt_llm.so"):
-        plugin_lib_path = "./tensorrt_llm/libs/" + "libnvinfer_plugin_tensorrt_llm.so"
+    if "linux" in platform:
+        lib_filename = "libnvinfer_plugin_tensorrt_llm.so"
+    else:
+        lib_filename = "libnvinfer_plugin_tensorrt_llm.dll"
+    plugin_lib_path = os.path.join("./tensorrt_llm/libs", lib_filename)
+    if os.path.exists(plugin_lib_path):
+        plugin_lib_path = "./tensorrt_llm/libs/" + plugin_lib_path
     else:
         try:
             import zipfile
@@ -858,9 +863,7 @@ def download_plugin_lib_path(py_version: str, platform: str) -> str:
             )
         with zipfile.ZipFile(file_name, "r") as zip_ref:
             zip_ref.extractall(".")  # Extract to a folder named 'tensorrt_llm'
-            plugin_lib_path = (
-                "./tensorrt_llm/libs/" + "libnvinfer_plugin_tensorrt_llm.so"
-            )
+            plugin_lib_path = "./tensorrt_llm/libs/" + plugin_lib_path
     return plugin_lib_path