Skip to content

Commit a6dcd31

Browse files
authored
Update test-mlperf-inference-abtf-poc.yml | Run docker from PR branch
1 parent 9924b6a commit a6dcd31

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.github/workflows/test-mlperf-inference-abtf-poc.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818
python-version: [ "3.8", "3.12" ]
1919
backend: [ "pytorch" ]
2020
implementation: [ "python" ]
21-
docker: [ "", " --docker --docker_mlc_repo=mlcommons@mlperf-automations --docker_mlc_repo_branch=dev --docker_dt" ]
21+
docker: [ "", " --docker --docker_mlc_repo=${{ github.event.pull_request.head.repo.html_url }} --docker_mlc_repo_branch=${{ github.event.pull_request.head.ref }} --docker_dt" ]
2222
extra-args: [ "--adr.compiler.tags=gcc", "--env.MLC_MLPERF_LOADGEN_BUILD_FROM_SRC=off" ]
2323
exclude:
2424
- os: ubuntu-24.04
@@ -28,16 +28,16 @@ jobs:
2828
- os: windows-latest
2929
extra-args: "--adr.compiler.tags=gcc"
3030
- os: windows-latest
31-
docker: " --docker --docker_mlc_repo=mlcommons@mlperf-automations --docker_mlc_repo_branch=dev --docker_dt"
31+
docker: " --docker --docker_mlc_repo=${{ github.event.pull_request.head.repo.html_url }} --docker_mlc_repo_branch=${{ github.event.pull_request.head.ref }} --docker_dt"
3232
# windows docker image is not supported in CM yet
3333
- os: macos-latest
3434
python-version: "3.8"
3535
- os: macos-13
3636
python-version: "3.8"
3737
- os: macos-latest
38-
docker: " --docker --docker_mlc_repo=mlcommons@mlperf-automations --docker_mlc_repo_branch=dev --docker_dt"
38+
docker: " --docker --docker_mlc_repo=${{ github.event.pull_request.head.repo.html_url }} --docker_mlc_repo_branch=${{ github.event.pull_request.head.ref }} --docker_dt"
3939
- os: macos-13
40-
docker: " --docker --docker_mlc_repo=mlcommons@mlperf-automations --docker_mlc_repo_branch=dev --docker_dt"
40+
docker: " --docker --docker_mlc_repo=${{ github.event.pull_request.head.repo.html_url }} --docker_mlc_repo_branch=${{ github.event.pull_request.head.ref }} --docker_dt"
4141

4242
steps:
4343
- uses: actions/checkout@v3

0 commit comments

Comments
 (0)