From 12e386f30804dbc2bdf6966dc7579e4c88160f2f Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Wed, 28 Jan 2026 17:14:44 -0800
Subject: [PATCH] Tune CUDA gaph sizes on B200 and H100 (#3077)

---
 mlx/backend/cuda/device.cpp | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/mlx/backend/cuda/device.cpp b/mlx/backend/cuda/device.cpp
index 12033d94..1d7ad7dc 100644
--- a/mlx/backend/cuda/device.cpp
+++ b/mlx/backend/cuda/device.cpp
@@ -214,13 +214,7 @@ std::pair<int, int> get_graph_limits(Device& d) {
       mb = 400;
       break;
     case 900: // H100
-      ops = 30;
-      mb = 400;
-      break;
     case 1000: // B200
-      ops = 50;
-      mb = 500;
-      break;
     case 1200: // Consumer Blackwell
       ops = 100;
       mb = 1000;