Skip to content

Commit

Permalink
Fix: Handle AWS T4 device without software version requirement
Browse files Browse the repository at this point in the history
  • Loading branch information
Only-bottle committed Feb 27, 2025
1 parent 70429ff commit 93ac86e
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions netspresso/benchmarker/v2/benchmarker.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,11 @@ def filter_device_by_version(
Returns:
Optional[DeviceInfo]: Filtered device info or None if no matching version
"""
# software_version이 None이고 device가 aws-t4인 경우 그대로 반환
if target_software_version is None and device.device_name == DeviceName.AWS_T4:
return device

# 기존 로직
filtered_versions = [
version for version in device.software_versions if version.software_version == target_software_version
]
Expand Down

0 comments on commit 93ac86e

Please sign in to comment.