Skip to content

Commit 10499af

Browse files
Merge pull request #299 from KernelTuner/fix-pycuda-cc
Support three-digit compute capability
2 parents a88c087 + c954f14 commit 10499af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernel_tuner/backends/pycuda.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def _finish_up():
106106
)
107107
if cc == "00":
108108
cc = self.context.get_device().compute_capability()
109-
self.cc = str(cc[0]) + str(cc[1])
109+
self.cc = str(cc)
110110
self.iterations = iterations
111111
self.current_module = None
112112
self.func = None

0 commit comments

Comments
 (0)