Thiago Crepaldi 418568d2e3 Add Float8 support to onnx exporter (#121281) ...

Fixes #106877

Pull Request resolved: https://github.com/pytorch/pytorch/pull/121281
Approved by: https://github.com/BowenBao, https://github.com/titaiwangms

2024-03-06 18:46:56 +00:00

..

assets

[ONNX] add test images to repo (#63717)

2021-08-23 07:43:21 -07:00

dynamo

[onnxrt, dynamo] Fix run with inputs on mix devices (#121159)

2024-03-04 23:39:33 +00:00

expect

[ONNX] Perform implicit casting of constants for the onnx::where operator (#118733) (#120619)

2024-03-04 19:27:30 +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

torch_export

[export][reland] Disable exported_program.__call__ (#120019)

2024-03-05 05:29:46 +00:00

autograd_helper.py

[ONNX] Format ONNX python with black

2022-05-05 00:19:22 +00:00

debug_embed_params.py

Apply ufmt linter to all py files under test/onnx (#81335)

2022-07-15 18:51:38 +00:00

error_reproduction.py

[ONNX] Support op consistency error reproduction (#119512)

2024-02-09 23:24:01 +00:00

onnx_test_common.py

[export][reland] Disable exported_program.__call__ (#120019)

2024-03-05 05:29:46 +00:00

pytorch_helper.py

[BE] f-stringify torch/ and scripts (#105538)

2023-07-21 19:35:24 +00:00

pytorch_test_common.py

[ONNX] Apply modularizarion to exported program exporting (#119498)

2024-02-09 22:57:42 +00:00

test_autograd_funs.py

[ONNX] Create common ExportTestCase base class (#88145)

2022-11-10 21:51:59 +00:00

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

Revert "Batch Norm Consolidation (#116092)"

2024-03-06 17:10:01 +00:00

test_fx_passes.py

[ONNX] Apply modularizarion to exported program exporting (#119498)

2024-02-09 22:57:42 +00:00

test_fx_to_onnx_decomp_skip.py

[ONNX][dynamo_export] Skip instance_norm decomp for export (#120866)

2024-03-01 23:51:16 +00:00

test_fx_to_onnx_with_onnxruntime.py

[export][reland] Disable exported_program.__call__ (#120019)

2024-03-05 05:29:46 +00:00

test_fx_to_onnx.py

Add Float8 support to onnx exporter (#121281)

2024-03-06 18:46:56 +00:00

test_fx_type_promotion.py

[ONNX][TypePromo] Materialize type promotion rules (#104063)

2023-07-01 01:20:08 +00:00

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

[ONNX] Add quantized model tests to CI (#80398)

2022-07-28 21:25:29 +00:00

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

[ONNX] Eliminate redundant TODOs (#119060)

2024-02-02 23:37:52 +00:00

test_onnxscript_runtime.py

[ONNX] Eliminate redundant TODOs (#119060)

2024-02-02 23:37:52 +00:00

test_op_consistency.py

[ONNX] Eliminate redundant TODOs (#119060)

2024-02-02 23:37:52 +00:00

test_operators.py

[BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027)

2023-12-20 19:35:08 +00:00

test_pytorch_jit_onnx.py

[ONNX] Perform implicit casting of constants for the onnx::where operator (#118733) (#120619)

2024-03-04 19:27:30 +00:00

test_pytorch_onnx_no_runtime.py

[ONNX][dynamo_export] Turn off opmath type promotion for div (#119112)

2024-02-07 03:27:00 +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

[ONNX] Perform implicit casting of constants for the onnx::where operator (#118733) (#120619)

2024-03-04 19:27:30 +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

[ONNX] Fix check_training_mode in symbolic_helper (#78376)

2022-05-27 00:38:16 +00:00

test_utility_funs.py

[TorchFix] Update old pretrained TorchVision API in tests (#111708)

2023-10-21 07:05:33 +00:00

test_verification.py

[ONNX] Properly skip tests by onnx version via 'unittest.skipIf' (#93316)

2023-02-01 01:14:07 +00:00

verify.py

Revert "[ONNX] Remove the depreacated function _export (#109763)"

2023-09-25 17:47:21 +00:00