pytorch/test/onnx
Justin Chu f3aba45049 [ONNX] Create onnxscript-torchlib specific xfails/skips for fx tests (#110536)
Creates xfail_onnxscript/skip_onnxscript so that it is clear torchlib needs to support it.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110536
Approved by: https://github.com/BowenBao
2023-10-05 00:39:05 +00:00
..
assets
dynamo [ONNX] Support torch.compile(backend="onnxrt", options=OrtBackendOptions(...)) (#107973) 2023-08-26 18:20:18 +00:00
expect [ONNX] Fix indexing issue of meshgrid op (#109350) 2023-09-15 19:49:43 +00:00
internal [ONNX] Set 'Generic[Diagnostic]' as base class for 'DiagnosticContext' (#107165) 2023-08-15 21:01:17 +00:00
model_defs removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
autograd_helper.py
debug_embed_params.py
onnx_test_common.py [ONNX] Bump onnx submodule to 1.14.1; ONNX Runtime 1.16 (#106984) 2023-08-28 20:11:29 +00:00
pytorch_helper.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
pytorch_test_common.py Add ONNX export support for huggingface's bigscience/bloom-560m model (#106930) 2023-08-11 16:34:24 +00:00
test_autograd_funs.py
test_custom_ops.py [ONNX] Expose OnnxRegistry publicly (#106140) 2023-08-04 20:46:03 +00:00
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 [ONNX] Create onnxscript-torchlib specific xfails/skips for fx tests (#110536) 2023-10-05 00:39:05 +00:00
test_fx_passes.py Change _dynamo.export to be export(f)(*args, **kwargs) (#106109) 2023-07-27 21:41:13 +00:00
test_fx_to_onnx.py [ONNX] Move large scale models without non-persistent buffers to runtime test (#108084) 2023-08-31 06:05:19 +00:00
test_fx_to_onnx_with_onnxruntime.py [ONNX] Move large scale models without non-persistent buffers to runtime test (#108084) 2023-08-31 06:05:19 +00:00
test_fx_type_promotion.py [ONNX][TypePromo] Materialize type promotion rules (#104063) 2023-07-01 01:20:08 +00:00
test_models.py
test_models_onnxruntime.py
test_models_quantized_onnxruntime.py
test_onnx_opset.py [ONNX] Export dynamic step size for aten::slice() (#104385) 2023-07-06 21:38:59 +00:00
test_onnxscript_no_runtime.py
test_onnxscript_runtime.py
test_op_consistency.py [ONNX] Return input itself for non-fp inputs and support decimals for aten::round op (#107920) 2023-08-26 05:54:52 +00:00
test_operators.py [ONNX] Fix indexing issue of meshgrid op (#109350) 2023-09-15 19:49:43 +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 [ONNX] Add initial support for FP8 ONNX export (#107962) 2023-09-08 20:40:39 +00:00
test_pytorch_onnx_onnxruntime.py torch->onnx export support: quantized::linear_relu (#109755) 2023-09-21 23:24:20 +00:00
test_pytorch_onnx_onnxruntime_cuda.py
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 [ONNX] Relax getting module attributes in ONNX export (#109759) 2023-09-23 02:47:51 +00:00
test_verification.py
verify.py Revert "[ONNX] Remove the depreacated function _export (#109763)" 2023-09-25 17:47:21 +00:00