From cb812c9832a3cc7f826c2c08ab1e14eefb13c14c Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 4 Mar 2024 20:54:26 +0000 Subject: [PATCH] Add windows constraint to mkl package in wheel (#121014) Follow up on: https://github.com/pytorch/pytorch/pull/102604 Address this comment: https://github.com/pytorch/pytorch/pull/102604#discussion_r1419944305 Whl metadata for all wheels published to pypi must match, otherwise poetry install will fail see this comment: https://github.com/pytorch/pytorch/issues/88049#issuecomment-1302555269 Pull Request resolved: https://github.com/pytorch/pytorch/pull/121014 Approved by: https://github.com/malfet --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 84cd3fd0403f986..770a1bba4b7fa51 100644 --- a/setup.py +++ b/setup.py @@ -1095,9 +1095,8 @@ def main(): "networkx", "jinja2", "fsspec", + 'mkl>=2021.1.1,<=2021.4.0; platform_system == "Windows"', ] - if IS_WINDOWS: - install_requires.append("mkl>=2021.1.1,<=2021.4.0") # Parse the command line and check the arguments before we proceed with # building deps and setup. We need to set values so `--help` works.