pytorch/test/onnx
Thiago Crepaldi 2a514f48d7 Add huggingface gpt2 fake tensor unit test for torch.onnx.dynamo_export (#115380)
open llama, dolly v2 and falcon are still broken regardless of `ExportedProgram`, so they were not moved from `test_fx_to_onnx.py` to `fx_to_onnx_onnxruntime.py`.

Dolly and falcon already have tracking issues, but a tracking issue was created for open llama: https://github.com/pytorch/pytorch/issues/115552

A tracking issue was created for `xfail_if_model_type_is_exportedprogram` and `xfail_if_model_type_is_not_exportedprogram` issues with unexpected success runs: https://github.com/pytorch/pytorch/issues/115747
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115380
Approved by: https://github.com/titaiwangms
2023-12-13 19:49:06 +00:00
..
assets
dynamo [ONNX] Relax unsupported node analysis on complex dtype (#113785) 2023-11-17 20:11:20 +00:00
expect Add decomp for replication_pad2d and use for CUDA deterministic (#111590) 2023-12-01 18:56:09 +00:00
internal
model_defs removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
torch_export Store user model to simplify ONNXProgram.{adapt_torch_*,__call__} APIs (#115281) 2023-12-09 07:46:12 +00:00
autograd_helper.py
debug_embed_params.py
onnx_test_common.py Move ONNX's TorchModelType to pytorch_test_common to fix circ. dep. (#115353) 2023-12-11 22:23:03 +00:00
pytorch_helper.py
pytorch_test_common.py Move ONNX's TorchModelType to pytorch_test_common to fix circ. dep. (#115353) 2023-12-11 22:23:03 +00:00
test_autograd_funs.py
test_custom_ops.py
test_export_modes.py Revert "[ONNX] Remove the depreacated function _export (#109763)" 2023-09-25 17:47:21 +00:00
test_fx_op_consistency.py Move ONNX's TorchModelType to pytorch_test_common to fix circ. dep. (#115353) 2023-12-11 22:23:03 +00:00
test_fx_passes.py Rename torch.onnx.ExportOutput* to ONNXProgram* (#112263) 2023-11-06 22:27:15 +00:00
test_fx_to_onnx_with_onnxruntime.py Add huggingface gpt2 fake tensor unit test for torch.onnx.dynamo_export (#115380) 2023-12-13 19:49:06 +00:00
test_fx_to_onnx.py [ONNX] Fix op level debug on complex dtype support (#114885) 2023-12-01 02:17:27 +00:00
test_fx_type_promotion.py
test_models_onnxruntime.py [TorchFix] Update old pretrained TorchVision API in tests (#111708) 2023-10-21 07:05:33 +00:00
test_models_quantized_onnxruntime.py
test_models.py [TorchFix] Update old pretrained TorchVision API in tests (#111708) 2023-10-21 07:05:33 +00:00
test_onnx_opset.py [ONNX] Refactor MaxPool to support dynamic inputs (#113318) 2023-11-10 23:23:49 +00:00
test_onnxscript_no_runtime.py
test_onnxscript_runtime.py
test_op_consistency.py [ONNX] Fix scalar type promotion between fp16 tensor and fp32 scalar (#113404) 2023-11-15 20:32:55 +00:00
test_operators.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_pytorch_jit_onnx.py [ONNX] Remove the depreacated function in symbolic_helper (#109681) 2023-09-20 19:31:39 +00:00
test_pytorch_onnx_no_runtime.py [BE]: ruff - enable PIE804 (#113951) 2023-11-17 21:23:02 +00:00
test_pytorch_onnx_onnxruntime_cuda.py [ATen] Support multi dim any and all reductions (#110310) 2023-10-24 21:33:53 +00:00
test_pytorch_onnx_onnxruntime.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_pytorch_onnx_shape_inference.py [ONNX] Remove the depreacated function in symbolic_helper (#109681) 2023-09-20 19:31:39 +00:00
test_symbolic_helper.py
test_utility_funs.py [TorchFix] Update old pretrained TorchVision API in tests (#111708) 2023-10-21 07:05:33 +00:00
test_verification.py
verify.py Revert "[ONNX] Remove the depreacated function _export (#109763)" 2023-09-25 17:47:21 +00:00