Skip to content

Commit 552f622

Browse files
committed
List skipped benchmarks
1 parent bf04936 commit 552f622

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

benchmarks/benchmark.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -392,8 +392,10 @@ async def benchmark_func_shared(data=data):
392392
else:
393393
print(f" Share_torch failed: {e}")
394394
failed_tests[test_name] = str(e)[:100]
395-
elif should_test_shared:
396-
# Extension was stopped due to previous error but should have been tested
395+
else:
396+
# Extension is None (either not created or was stopped)
397+
should_test_shared = torch_mode in ["both", "shared"]
398+
if should_test_shared:
397399
test_name = f"{name}_shared"
398400
skipped_tests[test_name] = "Extension stopped"
399401

0 commit comments

Comments
 (0)