cpu wheel builds for linux and mac

Reviewed By: nikhilaravi

Differential Revision: D20073426

fbshipit-source-id: fce83c9b63d630de1e6ebe2ff4790f29d11b65cc
This commit is contained in:
Jeremy Reizenstein
2020-02-24 13:36:02 -08:00
committed by Facebook Github Bot
parent 40be4cf78b
commit 4233c32887
5 changed files with 103 additions and 3 deletions

View File

@@ -14,7 +14,6 @@ import yaml
def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
w = []
# add "wheel" here for pypi
for btype in ["conda"]:
for python_version in ["3.6", "3.7", "3.8"]:
for cu_version in ["cu92", "cu100", "cu101"]:
@@ -26,6 +25,17 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
upload=upload,
filter_branch=filter_branch,
)
for btype in ["wheel"]:
for python_version in ["3.6", "3.7", "3.8"]:
for cu_version in ["cpu"]:
w += workflow_pair(
btype=btype,
python_version=python_version,
cu_version=cu_version,
prefix=prefix,
upload=upload,
filter_branch=filter_branch,
)
return indent(indentation, w)