Skip to content

Commit 289d8c6

Browse files
authored
Filter out operators we cannot test correctness for (#105)
1 parent befc4cd commit 289d8c6

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

BackendBench/scripts/dataset_filters.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,13 @@
2020
"_fft_c2c.default", # cuFFT only supports dimensions whose sizes are powers of two when computing in half precision
2121
]
2222

23+
UNTESTABLE_OPERATORS = [
24+
"empty_like", # We can check using metadata
25+
"new_empty", # We can check using metadata
26+
"new_empty_strided", # We can check using metadata
27+
"bernoulli", # We can write a custom test to verify this one (albeit not the randomness)
28+
]
29+
2330

2431
def apply_skip_ops_filter(ops):
2532
for op in ops:
@@ -28,6 +35,12 @@ def apply_skip_ops_filter(ops):
2835
op["why_excluded"].append("We cannot run this op on backendbench yet")
2936
op["runnable"] = False
3037

38+
if any(skip_op in op["op_name"] for skip_op in UNTESTABLE_OPERATORS):
39+
op["included_in_benchmark"] = False
40+
op["why_excluded"].append(
41+
"BackendBench does not support correctness testing for this op yet"
42+
)
43+
3144
if op["is_synthetic"]:
3245
op["included_in_benchmark"] = False
3346
op["why_excluded"].append(

0 commit comments

Comments
 (0)