Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 13 additions & 0 deletions BackendBench/scripts/dataset_filters.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,13 @@
"_fft_c2c.default", # cuFFT only supports dimensions whose sizes are powers of two when computing in half precision
]

UNTESTABLE_OPERATORS = [
"empty_like", # We can check using metadata
"new_empty", # We can check using metadata
"new_empty_strided", # We can check using metadata
"bernoulli", # We can write a custom test to verify this one (albeit not the randomness)
]


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

if any(skip_op in op["op_name"] for skip_op in UNTESTABLE_OPERATORS):
op["included_in_benchmark"] = False
op["why_excluded"].append(
"BackendBench does not support correctness testing for this op yet"
)

if op["is_synthetic"]:
op["included_in_benchmark"] = False
op["why_excluded"].append(
Expand Down