Skip to content

Commit fe0cfc7

Browse files
authored
Update to make notebook_list.py compatible with numba 0.53 (#1455)
A recent update to numba 0.53 in CI broke this script and caused CI failures. This makes the script compatible with both pre and post numba 0.53 versions. Tested in a local env with numba 0.53 installed. Authors: - Rick Ratzel (@rlratzel) Approvers: - Alex Fender (@afender) - Brad Rees (@BradReesWork) - Chuck Hastings (@ChuckHastings) - AJ Schmidt (@ajschmidt8) URL: #1455
1 parent 591f3fb commit fe0cfc7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ci/gpu/notebook_list.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@
2424
pascal = False
2525

2626
device = cuda.get_current_device()
27-
cc = getattr(device, 'COMPUTE_CAPABILITY')
27+
# check for the attribute using both pre and post numba 0.53 names
28+
cc = getattr(device, 'COMPUTE_CAPABILITY', None) or \
29+
getattr(device, 'compute_capability')
2830
if (cc[0] < 7):
2931
pascal = True
3032

0 commit comments

Comments
 (0)