Skip to content

Commit

Permalink
Attempt to fix the Python wheels for Windows (#3979)
Browse files Browse the repository at this point in the history
  • Loading branch information
marbre authored Jan 22, 2025
1 parent 2fb7d6e commit 9dd94fb
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@
# that here, and just package up its contents.
import os
import pathlib
import platform
import shutil
import subprocess
import sys
Expand Down Expand Up @@ -202,6 +203,9 @@ def run(self):
torch_mlir_opt_dst = os.path.join(
target_dir, "torch_mlir", "_mlir_libs", "torch-mlir-opt"
)
if platform.system() == "Windows":
torch_mlir_opt_src += ".exe"
torch_mlir_opt_dst += ".exe"
shutil.copy2(torch_mlir_opt_src, torch_mlir_opt_dst, follow_symlinks=False)


Expand Down

0 comments on commit 9dd94fb

Please sign in to comment.