hopefully fix CI

This commit is contained in:
harisreedhar
2026-02-12 21:11:00 +05:30
committed by henryruhs
parent ce0f989c28
commit b39af57f22
+6 -5
View File
@@ -113,12 +113,13 @@ def resolve_cache_path() -> str:
def resolve_cudnn_conv_algo_search(execution_providers : List[ExecutionProvider]) -> str:
graphic_devices = detect_static_graphic_devices(tuple(execution_providers))
product_names = ('GeForce GTX 1630', 'GeForce GTX 1650', 'GeForce GTX 1660')
if has_execution_provider('cuda') or has_execution_provider('tensorrt'):
graphic_devices = detect_static_graphic_devices(tuple(execution_providers))
product_names = ('GeForce GTX 1630', 'GeForce GTX 1650', 'GeForce GTX 1660')
for graphic_device in graphic_devices:
if graphic_device.get('product').get('name').startswith(product_names):
return 'DEFAULT'
for graphic_device in graphic_devices:
if graphic_device.get('product').get('name').startswith(product_names):
return 'DEFAULT'
return 'EXHAUSTIVE'