diff --git a/.github/workflows/test_e2eshark.yml b/.github/workflows/test_e2eshark.yml index 81cfce9b..a0394138 100644 --- a/.github/workflows/test_e2eshark.yml +++ b/.github/workflows/test_e2eshark.yml @@ -20,97 +20,97 @@ jobs: fail-fast: false matrix: include: - # - name: mi300_gpu1_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: nlp-shard1 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu2_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: nlp-shard2 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu3_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: nlp-shard3 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu4_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: shark-test-suite - # cache-dir: /home/sai/shark-test-suite-models-cache + - name: mi300_gpu1_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: nlp-shard1 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu2_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: nlp-shard2 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu3_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: nlp-shard3 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu4_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: shark-test-suite + cache-dir: /data/e2eshark/shark-test-suite-models-cache # - name: mi300_gpu5_test # runs-on: nodai-amdgpu-mi300-x86-64 # backend: rocm # device: hip # target-chip: gfx942 # test-file: vai-hf-cnn-fp32-shard1 - # cache-dir: /home/sai/shark-test-suite-models-cache + # cache-dir: /data/e2eshark/shark-test-suite-models-cache # - name: mi300_gpu6_test # runs-on: nodai-amdgpu-mi300-x86-64 # backend: rocm # device: hip # target-chip: gfx942 # test-file: vai-hf-cnn-fp32-shard2 - # cache-dir: /home/sai/shark-test-suite-models-cache + # cache-dir: /data/e2eshark/shark-test-suite-models-cache # - name: mi300_gpu7_test # runs-on: nodai-amdgpu-mi300-x86-64 # backend: rocm # device: hip # target-chip: gfx942 # test-file: vai-hf-cnn-fp32-shard3 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu8_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: vai-int8-p0p1-shard1 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu9_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: vai-int8-p0p1-shard2 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu10_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: vai-int8-p0p1-shard3 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu11_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: vai-vision-int8 - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu12_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # device: hip - # target-chip: gfx942 - # test-file: migraphx - # cache-dir: /home/sai/shark-test-suite-models-cache - # - name: mi300_gpu13_test - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: hip - # device: hip - # target-chip: gfx942 - # test-file: onnxrt-iree-ep - # cache-dir: /home/sai/shark-test-suite-models-cache + # cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu8_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: vai-int8-p0p1-shard1 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu9_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: vai-int8-p0p1-shard2 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu10_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: vai-int8-p0p1-shard3 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu11_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: vai-vision-int8 + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu12_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + device: hip + target-chip: gfx942 + test-file: migraphx + cache-dir: /data/e2eshark/shark-test-suite-models-cache + - name: mi300_gpu13_test + runs-on: nodai-amdgpu-mi300-x86-64 + backend: hip + device: hip + target-chip: gfx942 + test-file: onnxrt-iree-ep + cache-dir: /data/e2eshark/shark-test-suite-models-cache - name: cpu_shard1_test runs-on: nodai-amdgpu-mi250-x86-64 backend: llvm-cpu @@ -328,12 +328,12 @@ jobs: max-parallel: 1 matrix: include: - # - name: merge_rocm_reports - # runs-on: nodai-amdgpu-mi300-x86-64 - # backend: rocm - # regression-blob: rocm + - name: merge_rocm_reports + runs-on: nodai-amdgpu-mi300-x86-64 + backend: rocm + regression-blob: rocm - name: merge_cpu_reports - runs-on: nodai-amdgpu-mi250-x86-64 + runs-on: nodai-amdgpu-mi300-x86-64 backend: llvm-cpu regression-blob: cpu env: