diff --git a/.buildkite/test_areas/torch_abi.yaml b/.buildkite/test_areas/torch_abi.yaml
index 619910374f..9e8375587a 100644
--- a/.buildkite/test_areas/torch_abi.yaml
+++ b/.buildkite/test_areas/torch_abi.yaml
@@ -1,11 +1,10 @@
 group: Torch ABI
 depends_on:
-  - image-build-cpu
+  - image-build
 steps:
 - label: ":nvidia: (L4) Torch Stable ABI Audit"
   key: torch-stable-abi-audit
   timeout_in_minutes: 5
-  device: cpu-small
   source_file_dependencies:
   - .buildkite/check-torch-abi.py
   - csrc/