Support three-digit compute capability

csbnw · csbnw · commit 018a961fd14e · 2025-05-01T08:04:19.000+02:00
E.g. 120, for Blackwell.
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def _finish_up():`
`106`	`106`	`)`
`107`	`107`	`if cc == "00":`
`108`	`108`	`cc = self.context.get_device().compute_capability()`
`109`		`- self.cc = str(cc[0]) + str(cc[1])`
	`109`	`+ self.cc = str(cc)`
`110`	`110`	`self.iterations = iterations`
`111`	`111`	`self.current_module = None`
`112`	`112`	`self.func = None`