Update build-test-rocm.sh

ROCm · Jun 14, 2024 · eb71ffa · eb71ffa
1 parent b879b3e
commit eb71ffa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/scripts/build-test-rocm.sh b/.github/scripts/build-test-rocm.sh
@@ -30,8 +30,8 @@ build_and_test() {
 setup_container() {
     image=rocm/pytorch:latest
     echo "Using image $image"
-    trap 'cleanup' ERR
     docker run --platform "linux/$build_arch" -i -d -w /src --device=/dev/kfd --device=/dev/dri --group-add video --name bnb_rocm_test "$image"
+    trap 'cleanup' ERR
     docker cp $PWD bnb_rocm_test:/src
 }