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
This commit is contained in:
atalman
2024-03-04 20:54:26 +00:00
committed by PyTorch MergeBot
parent 4cdc2d7096
commit cb812c9832

View File

@ -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.