Skip to content

Commit b86ee7f

Browse files
authored
add H100 and L4 (#85)
Co-authored-by: John Calderon <[email protected]>
1 parent 021127e commit b86ee7f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

deepview_profile/analysis/session.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -306,6 +306,8 @@ def habitat_predict(self):
306306
habitat.Device.A40,
307307
habitat.Device.A4000,
308308
habitat.Device.RTX4000,
309+
habitat.Device.L4,
310+
habitat.Device.H100
309311
]
310312

311313
# Detect source GPU

0 commit comments

Comments
 (0)