saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-09 17:28:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test
History
Tianlei Wu 7c3a25225f
[CUDA] update test_flash_attn_cuda.py for Windows (#21006) 
Currently test_flash_attn_cuda.py can only run in Linux. It is because
it uses triton for rotary reference implementation, and triton python
package is not available in Windows.

This changes the script to allow the test run in Windows, so that we can
test memory efficient attention in Windows.

Due to limitation, rotary is excluded in testing on Windows.
2024-06-13 12:50:02 -07:00
..
common Fully dynamic ETW controlled logging for ORT and QNN logs (#20537) 2024-06-06 21:11:14 -07:00
contrib_ops relax seq len checking in rotary_emb (#20778) 2024-06-08 18:39:06 +08:00
cuda_host Enable CUDA EP unit testing on Windows (#20039) 2024-03-27 13:32:36 -07:00
custom_op_registration
debug_node_inputs_outputs Fix compiler error when onnxruntime_DEBUG_NODE_INPUTS_OUTPUTS is enabled (#20889) 2024-05-31 18:07:53 -07:00
flatbuffers Support >2GB of Tensor data in training checkpoint (#20077) 2024-04-22 15:17:43 -07:00
framework Avoid call to Node::ToProto on first Graph::Resolve to improve session creation performance. (#20296) 2024-04-17 10:07:12 +10:00
fuzzing
global_thread_pools
ir Fix missing node during mem efficient topo sort (#20497) 2024-05-06 17:25:23 +08:00
logging_apis
mlas Fix bench_sqnbitgemm.cpp benchmark argument name list. (#20858) 2024-05-30 18:59:54 -07:00
onnx [CPU EP] Add blocked quantization to QuantizeLinear op kernel (#20977) 2024-06-11 20:25:28 -07:00
opaque_api
optimizer [CPU EP] Int4 support for QuantizeLinear, DequantizeLinear, and Transpose (#20362) 2024-05-30 18:56:24 -07:00
perftest [TensorRT EP] Support engine hardware compatibility (#20669) 2024-05-28 18:12:56 -07:00
platform Remove deprecated "mobile" packages (#20941) 2024-06-07 16:20:32 -05:00
proto
providers Adding a sm80 q4 gemm kernel for small tiles (#20545) 2024-06-12 16:02:26 -07:00
python [CUDA] update test_flash_attn_cuda.py for Windows (#21006) 2024-06-13 12:50:02 -07:00
quantization
shared_lib [CPU EP] Int4 support for QuantizeLinear, DequantizeLinear, and Transpose (#20362) 2024-05-30 18:56:24 -07:00
testdata Support loading from model with multiple QNN context binary (#20930) 2024-06-06 14:44:57 -07:00
unittest_main [CUDA] upgrade cutlass to 3.5.0 (#20940) 2024-06-11 13:32:15 -07:00
util [CPU EP] Int4 support for QuantizeLinear, DequantizeLinear, and Transpose (#20362) 2024-05-30 18:56:24 -07:00
wasm
win_getopt
xctest
run_benchmark.py Add onnxruntime/test/run_benchmark.py helper script. (#19234) 2024-04-05 07:02:01 -07:00
run_benchmark.readme.md Add onnxruntime/test/run_benchmark.py helper script. (#19234) 2024-04-05 07:02:01 -07:00