mathieu/ai-suite-rocm-local
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

try to fix env to workflow

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-28 15:29:51 +02:00
parent 12b41b2657
commit 1435cf6775
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/build-and_publish-prebuilts-whl.yml vendored
View File

@ -31,6 +31,9 @@ jobs:
# Step 3: Build the Docker image # Step 3: Build the Docker image
- name: Build Docker image - name: Build Docker image
run: | run: |
echo ${{ inputs.rocm_version }}
echo ${{ inputs.torch_version }}
docker build \ docker build \
--build-arg ROCM_VERSION=${{ inputs.rocm_version }} \ --build-arg ROCM_VERSION=${{ inputs.rocm_version }} \
--build-arg TORCH_VERSION=${{ inputs.torch_version }} \ --build-arg TORCH_VERSION=${{ inputs.torch_version }} \