mirror of https://github.com/llvm/torch-mlir
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
# See https://llvm.org/LICENSE.txt for license information.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
# Also available under a BSD-style license. See LICENSE.
|
|
|
|
# This file describes the sets of tests expected to fail for each config.
|
|
# This information is deliberately kept in a side table, rather than
|
|
# in-situ on the test, as a deliberate layering decision: tests should
|
|
# have unique keys to identify them and enable side tables of various kinds
|
|
# (this includes down into lower parts of the stack, where a side table
|
|
# might be used to keep more elaborate sets of testing configurations).
|
|
|
|
# Lists of tests that fail to even reach the backends.
|
|
# These represent further work needed in torch-mlir to lower them properly
|
|
# to the backend contract.
|
|
COMMON_TORCH_MLIR_LOWERING_XFAILS = {
|
|
"QuantizedMLP_basic",
|
|
"IouOfModule_basic",
|
|
}
|
|
# Fails due to https://github.com/llvm/torch-mlir/issues/448
|
|
SIZE_ZERO_TENSOR_XFAILS = {
|
|
"SliceEndSleStartModule_basic",
|
|
"SliceStartEqEndModule_basic",
|
|
"SliceOutOfUpperBoundIndexModule_basic",
|
|
}
|
|
REFBACKEND_XFAIL_SET = set.union(COMMON_TORCH_MLIR_LOWERING_XFAILS, SIZE_ZERO_TENSOR_XFAILS)
|
|
|
|
# Write the TOSA set as a "passing" set as it is very early in development
|
|
# and very few tests work yet.
|
|
TOSA_PASS_SET = {
|
|
"ElementwiseUnaryModule_basic",
|
|
"ElementwiseBinaryModule_basic",
|
|
"ElementwiseSigmoidModule_basic",
|
|
"ElementwiseReluModule_basic",
|
|
"ElementwiseFloorModule_basic",
|
|
"ElementwiseLogModule_basic",
|
|
"TanhBackward_basic",
|
|
"ElementwiseAddModule_basic",
|
|
"ReturnThreeTensorFloat32_basic",
|
|
"AddCMulModule_basic",
|
|
"AddCDivModule_basic",
|
|
"SqueezeModule_broadcast",
|
|
}
|