Switch run_iree_dispatch_profiler to use install-dir. (#16327)
Progress on https://github.com/openxla/iree/issues/16203
Test run triggered at
https://github.com/openxla/iree/actions/runs/7792358411
skip-ci: not covered by presubmit CI
diff --git a/.github/workflows/run_iree_dispatch_profiler.yml b/.github/workflows/run_iree_dispatch_profiler.yml
index 29534f8..0579f00 100644
--- a/.github/workflows/run_iree_dispatch_profiler.yml
+++ b/.github/workflows/run_iree_dispatch_profiler.yml
@@ -54,24 +54,24 @@
RESULTS_DIR: dispatch-profiler-results-cuda
GCS_UPLOAD_PARENT_DIR: "gs://dispatch-profiler-artifacts/cuda"
GCS_UPLOAD_DIR_NAME: ${{ needs.setup.outputs.artifact-upload-dir }}
- BUILD_DIR: ${{ needs.build_all.outputs.build-dir }}
- BUILD_DIR_ARCHIVE: ${{ needs.build_all.outputs.build-dir-archive }}
- BUILD_DIR_GCS_ARTIFACT: ${{ needs.build_all.outputs.build-dir-gcs-artifact }}
+ INSTALL_DIR: ${{ needs.build_all.outputs.install-dir }}
+ INSTALL_DIR_ARCHIVE: ${{ needs.build_all.outputs.install-dir-archive }}
+ INSTALL_DIR_GCS_ARTIFACT: ${{ needs.build_all.outputs.install-dir-gcs-artifact }}
IREE_SHA: ${{ github.sha }}
steps:
- name: "Checking out repository"
uses: actions/checkout@8f4b7f84864484a7bf31766abe9204da3cbe65b3 # v3.5.0
- - name: "Downloading build dir archive"
- run: gcloud storage cp "${BUILD_DIR_GCS_ARTIFACT}" "${BUILD_DIR_ARCHIVE}"
- - name: "Extracting install from build dir archive"
- run: tar -vxf "${BUILD_DIR_ARCHIVE}" "${BUILD_DIR}/install"
+ - name: "Downloading install dir archive"
+ run: gcloud storage cp "${INSTALL_DIR_GCS_ARTIFACT}" "${INSTALL_DIR_ARCHIVE}"
+ - name: "Extracting install directory"
+ run: tar -vxf "${INSTALL_DIR_ARCHIVE}"
- name: "Running IREE Dispatch Profiler on CUDA"
run: |
mkdir "${RESULTS_DIR}"
./build_tools/github_actions/docker_run.sh \
--gpus all \
gcr.io/iree-oss/nvidia-bleeding-edge@sha256:2eb17e2e8e0d5d25600e667e3b7a71e8d39d9d12a99ec2ad6bdabbc8919db43b \
- ./experimental/dispatch_profiler/profile_all.sh "${BUILD_DIR}/install/bin" \
+ ./experimental/dispatch_profiler/profile_all.sh "${INSTALL_DIR}/bin" \
"${RESULTS_DIR}"
- name: "Uploading results"
run: |