We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
09-persistent-matmul.py
1 parent 57083c7 commit 35023b0Copy full SHA for 35023b0
python/tutorials/09-persistent-matmul.py
@@ -77,7 +77,7 @@ def supports_tma():
77
78
79
def is_hopper():
80
- return torch.cuda.get_device_capability()[0] == 9
+ return is_cuda() and torch.cuda.get_device_capability()[0] == 9
81
82
83
def supports_ws():
0 commit comments