From 39c2cc523dd621010d84b1582022024b897e0256 Mon Sep 17 00:00:00 2001
From: Matt Wong <156021403+mawong-amd@users.noreply.github.com>
Date: Fri, 19 Jul 2024 13:02:10 +0000
Subject: [PATCH] Use 16gb shm for container

---
 .buildkite/run-amd-test.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh
index b459b1097e5fa..618d712b0279b 100644
--- a/.buildkite/run-amd-test.sh
+++ b/.buildkite/run-amd-test.sh
@@ -73,6 +73,7 @@ HF_MOUNT="/root/.cache/huggingface"
 docker run \
         --device /dev/kfd --device /dev/dri \
         --network host \
+        --shm-size=16gb \
         --rm \
         -e HF_TOKEN \
         -v ${HF_CACHE}:${HF_MOUNT} \