Skip to content

Commit

Permalink
Update wheels-generator.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
OutofAi authored Aug 6, 2024
1 parent 0e8c9f9 commit 5258aae
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/wheels-generator.yml
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ jobs:
$TORCH_VERSION_NO_DOT = $TORCH_VERSION -replace '\.', ''
cd bindings/torch/dist
Get-ChildItem -Name | ForEach-Object {
$new_filename = $_ -replace '1.7', "1.7+arch${{ matrix.arch }}+torch$TORCH_VERSION_NO_DOT+cuda$CUDA_VERSION_NO_DOT"
$new_filename = $_ -replace '1.7', "1.7.post{{ matrix.arch }}$TORCH_VERSION_NO_DOT$CUDA_VERSION_NO_DOT"
Rename-Item -Path $_ -NewName $new_filename
echo "Uploading $new_filename"
gh release upload ${{ env.TINY_CUDA_NN_VERSION }} "$new_filename" --clobber --repo ${{ github.repository }}
Expand Down Expand Up @@ -215,7 +215,7 @@ jobs:
cd bindings/torch/dist
for filename in *; do
# Rename the file by inserting .post${{ matrix.arch }} directly after "1.7"
new_filename=$(echo "$filename" | sed "s/1.7/1.7+arch${{ matrix.arch }}+torch${{ matrix.torchversion }}+cuda${{ matrix.cudaversion }}/")
new_filename=$(echo "$filename" | sed "s/1.7/1.7.post${{ matrix.arch }}${{ matrix.torchversion }}${{ matrix.cudaversion }}/")
mv "$filename" "$new_filename"
echo "Uploading $new_filename"
gh release upload ${{ env.TINY_CUDA_NN_VERSION }} "$new_filename" --clobber --repo ${{ github.repository }}
Expand Down

0 comments on commit 5258aae

Please sign in to comment.