From 018a961fd14ef9e19caf868ab105e222f8d17dc4 Mon Sep 17 00:00:00 2001
From: Bram Veenboer <bram.veenboer@gmail.com>
Date: Thu, 1 May 2025 08:04:19 +0200
Subject: [PATCH] Support three-digit compute capability

E.g. 120, for Blackwell.
---
 kernel_tuner/backends/pycuda.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kernel_tuner/backends/pycuda.py b/kernel_tuner/backends/pycuda.py
index 7fddc9393..81d8de67d 100644
--- a/kernel_tuner/backends/pycuda.py
+++ b/kernel_tuner/backends/pycuda.py
@@ -106,7 +106,7 @@ def _finish_up():
         )
         if cc == "00":
             cc = self.context.get_device().compute_capability()
-        self.cc = str(cc[0]) + str(cc[1])
+        self.cc = str(cc)
         self.iterations = iterations
         self.current_module = None
         self.func = None