We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 0e1b0e6 + ae2388e commit ffae748Copy full SHA for ffae748
onnxscript/tests/function_libs/torch_lib/ops_test_common.py
@@ -161,6 +161,10 @@ def add_decorate_info(
161
ops_mapping = {(info.name, info.variant_test_name): info for info in all_opinfos}
162
for decorate_meta in skip_or_xfails:
163
opinfo = ops_mapping.get((decorate_meta.op_name, decorate_meta.variant_name))
164
+ if opinfo is None and not decorate_meta.enabled_if:
165
+ # If the OpInfo doesn't exist and it is not enabled, we skip the OpInfo
166
+ # because it could be an OpInfo that is in torch-nightly but not older versions.
167
+ continue
168
assert (
169
opinfo is not None
170
), f"Couldn't find OpInfo for {decorate_meta}. Did you need to specify variant_name?"
0 commit comments