llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[CI] Enable the tests for fx_importer in the CI (#3168) 

Replace the torchdynamo e2e with the fx_importer e2e
pull/3163/head
penguin_wwy 2024-04-16 12:20:23 +08:00 committed by GitHub
parent ae4724763a
commit 10b6062d41
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
build_tools/ci/test_posix.sh
View File

@ -37,10 +37,9 @@ case $torch_version in
# python -m e2e_testing.main --config=lazy_tensor_core -v # python -m e2e_testing.main --config=lazy_tensor_core -v
# echo "::endgroup::" # echo "::endgroup::"
# TODO: There is one failing test in this group on stable. It could # TODO: Need to verify in the stable version
# be xfailed vs excluding entirely. echo "::group::Run FxImporter e2e integration tests"
echo "::group::Run TorchDynamo e2e integration tests" python -m e2e_testing.main --config=fx_importer -v
python -m e2e_testing.main --config=torchdynamo -v
echo "::endgroup::" echo "::endgroup::"
;; ;;
stable) stable)

6
build_tools/python_deploy/build_linux_packages.sh
View File

@ -308,10 +308,8 @@ function test_in_tree() {
echo ":::: Run Onnx e2e integration tests" echo ":::: Run Onnx e2e integration tests"
python -m e2e_testing.main --config=onnx -v python -m e2e_testing.main --config=onnx -v
# Dynamo is changing a lot in nightly versions, and thus the implementation echo ":::: Run FxImporter e2e integration tests"
# tends to become incompatible to the stable version. python -m e2e_testing.main --config=fx_importer -v
echo ":::: Run TorchDynamo e2e integration tests"
python -m e2e_testing.main --config=torchdynamo -v
;; ;;
stable) stable)
echo ":::: Test with stable torch" echo ":::: Test with stable torch"

3
projects/pt1/e2e_testing/main.py
View File

@ -164,6 +164,9 @@ def main():
# Report the test results. # Report the test results.
failed = report_results(results, xfail_set, args.verbose, args.config) failed = report_results(results, xfail_set, args.verbose, args.config)
if args.config == "torchdynamo":
print("\033[91mWarning: the TorchScript based dynamo support is deprecated. "
"The config for torchdynamo is planned to be removed in the future.\033[0m")
if args.ignore_failures: if args.ignore_failures:
sys.exit(0) sys.exit(0)
sys.exit(1 if failed else 0) sys.exit(1 if failed else 0)

7
projects/pt1/e2e_testing/xfail_sets.py
View File

@ -401,6 +401,12 @@ FX_IMPORT_XFAIL_SET = {
"AtenIntTensorCharDtypeModule_basic", "AtenIntTensorCharDtypeModule_basic",
"AtenItemFpOpModule_basic", "AtenItemFpOpModule_basic",
"AtenItemIntOpModule_basic", "AtenItemIntOpModule_basic",
"AtenMmQint8_basic",
"AtenMatmulQint8_basic",
"AtenMatmulQint8MV_basic",
"AtenMmQMixedSigni8_basic",
"AtenMatmulQMixedSigni8_basic",
"AtenMatmulQMixedSigni8Transpose_basic",
"AtenMmQuint8_basic", "AtenMmQuint8_basic",
"AtenSubFloatModule_basic", "AtenSubFloatModule_basic",
"BincountMinlengthModule_basic", "BincountMinlengthModule_basic",
@ -474,6 +480,7 @@ FX_IMPORT_XFAIL_SET = {
"QuantizedMLP_basic", "QuantizedMLP_basic",
"QuantizedNoLayer_basic", "QuantizedNoLayer_basic",
"QuantizedSingleLayer_basic", "QuantizedSingleLayer_basic",
"QuantizedBatchedInputSingleLayer_basic",
"ReduceMaxAlongDimUnsignedInt_basic", "ReduceMaxAlongDimUnsignedInt_basic",
"ReduceMinAlongDimUnsignedInt_basic", "ReduceMinAlongDimUnsignedInt_basic",
"RsubInt0d_NumToTensor_Module_basic", "RsubInt0d_NumToTensor_Module_basic",