Skip to content

Commit 809cb26

Browse files
authored
Use A100 for LLama2 model test (#21068)
### Description ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
1 parent 9ef4f1b commit 809cb26

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,10 @@ stages:
296296
skipComponentGovernanceDetection: true
297297
workspace:
298298
clean: all
299-
pool: Onnxruntime-Linux-A10-24G
299+
pool:
300+
name: Onnxruntime-Linux-GPU-A100-WUS3
301+
demands:
302+
- WorkFolder -equals /mnt/storage/
300303
steps:
301304
- task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3
302305
displayName: 'Clean Agent Directories'

0 commit comments

Comments
 (0)