saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-19 02:03:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test/python
History
Changming Sun b7ef81a034
Move Linux GPU CI pipeline to A10 (#23235) 
Move Linux GPU CI pipeline to A10 machines which are more advanced.
Retire onnxruntime-Linux-GPU-T4 machine pool.
Disable run_lean_attention test because the new machines do not have
enough shared memory.

```
skip loading trt attention kernel fmha_mhca_fp16_128_256_sm86_kernel because no enough shared memory
[E:onnxruntime:, sequential_executor.cc:505 ExecuteKernel] Non-zero status code returned while running MultiHeadAttention node. Name:'MultiHeadAttention_0' Status Message: CUDA error cudaErrorInvalidValue:invalid argument
```
2025-01-04 19:11:37 -08:00
..
contrib_ops
quantization [Quant tool] Handle input models with pre-quantized weights (#22633) 2024-11-14 13:48:46 -08:00
testdata
transformers Move Linux GPU CI pipeline to A10 (#23235) 2025-01-04 19:11:37 -08:00
helper.py
onnx_backend_test_series.py Revert DML pipeline changes (#23135) 2024-12-18 10:42:10 -08:00
onnxruntime_test_collective.py
onnxruntime_test_distributed.py
onnxruntime_test_engine_wrapper.py
onnxruntime_test_float8.py
onnxruntime_test_float8_gemm8.py
onnxruntime_test_python.py Support onnx data types (bfloat16, float8) in python I/O binding APIs (#22306) 2024-10-04 17:29:15 -07:00
onnxruntime_test_python_azure.py
onnxruntime_test_python_backend.py
onnxruntime_test_python_backend_mlops.py Ignore ruff rule N813 (#21477) 2024-07-24 17:48:22 -07:00
onnxruntime_test_python_cudagraph.py
onnxruntime_test_python_dmlgraph.py
onnxruntime_test_python_iobinding.py Support onnx data types (bfloat16, float8) in python I/O binding APIs (#22306) 2024-10-04 17:29:15 -07:00
onnxruntime_test_python_keras.py
onnxruntime_test_python_mlops.py Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
onnxruntime_test_python_nested_control_flow_op.py
onnxruntime_test_python_sparse_matmul.py Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
onnxruntime_test_python_symbolic_shape_infer.py Re-enable test symbolic shape infer (#22737) 2024-11-14 11:28:00 -08:00
onnxruntime_test_scatternd.py
requirements.txt Integrate onnx 1.17.0 (#21897) 2024-12-24 09:02:02 -08:00
test_pytorch_export_contrib_ops.py