Skip to content

Commit

Permalink
Fix cu100 smokes
Browse files Browse the repository at this point in the history
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
  • Loading branch information
ezyang committed Aug 6, 2019
1 parent c2c2022 commit 50b4402
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions smoke_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -56,11 +56,12 @@ elif [[ "$DESIRED_CUDA" == 'cpu' && "$(uname)" != 'Darwin' ]]; then
else
package_name='pytorch-nightly'
fi
if [[ "$(uname)" == 'Darwin' ]]; then
if [[ "$(uname)" == 'Darwin' ]] || [[ "$DESIRED_CUDA" == "cu100" ]]; then
package_name_and_version="${package_name}==${NIGHTLIES_DATE_PREAMBLE}${DATE}"
else
# Linux binaries have the cuda version appended to them. This is only on
# linux, since all macos builds are cpu.
# linux, since all macos builds are cpu. (NB: We also omit
# DESIRED_CUDA if it's the default)
package_name_and_version="${package_name}==${NIGHTLIES_DATE_PREAMBLE}${DATE}+${DESIRED_CUDA}"
fi

Expand Down

0 comments on commit 50b4402

Please sign in to comment.