diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index 7faa61fd59..af027ff9c7 100755
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -1322,7 +1322,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
         if DEVICE_TYPE == "cuda":
             # CUDA-specific options (added to base options)
             cuda_options = """
-            "triton.enable_persistent_tma_matmul": torch.cuda.get_device_capability()[0] == 9,"""
+            "triton.enable_persistent_tma_matmul": torch.cuda.get_device_capability()[0] >= 9,"""
             # cutlass options were added in PyTorch 2.8.0
             if torch_version >= Version("2.8.0"):
                 cuda_options += """