Skip to content

Commit 76885c0

Browse files
committed
arch: Warn instead of fail if get_nvidia_cc not behaves
1 parent 2ea79d4 commit 76885c0

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

devito/arch/archinfo.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1023,7 +1023,12 @@ def supports(self, query, language=None):
10231023
return False
10241024

10251025
cc = get_nvidia_cc()
1026-
if query == 'async-loads' and cc >= 80:
1026+
if cc is None:
1027+
# Something off with `get_nvidia_cc` on this system
1028+
warning(f"Couldn't establish if `query={query}` is supported on this "
1029+
"system. Assuming it is not.")
1030+
return False
1031+
elif query == 'async-loads' and cc >= 80:
10271032
# Asynchronous pipeline loads -- introduced in Ampere
10281033
return True
10291034
elif query in ('tma', 'thread-block-cluster') and cc >= 90:

0 commit comments

Comments
 (0)