Skip to content

Commit 5707666

Browse files
authored
fix: call torch.cuda.is_available() in available_devices (#2602)
1 parent 6720e82 commit 5707666

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

thunder/core/devices.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def available_devices() -> tuple[Device]:
140140
available_devices = [cpu]
141141

142142
# Short-circuits if there are no CUDA devices
143-
if not torch.cuda.is_available:
143+
if not torch.cuda.is_available():
144144
return available_devices
145145

146146
# NOTE torch.cuda.is_available, extends with CUDA devices

0 commit comments

Comments
 (0)