Repository navigation
feat: add OpenVINO runtime and all RF-DETR detection sizes - #16
Open
leeclemnet wants to merge 3 commits into
Open
leeclemnet wants to merge 3 commits into
leeclemnet wants to merge 3 commits into
Conversation
|
Review the following changes in direct dependencies. Learn more about Socket for GitHub.
|
leeclemnet
force-pushed
the
feat/cpu-runtimes
branch
2 times, most recently
from
September 29, 2026 20:34
ea209a5 to
29dfcf5
Compare
This was referenced Sep 29, 2026
leeclemnet
force-pushed
the
feat/cpu-runtimes
branch
from
September 29, 2026 20:41
29dfcf5 to
d955761
Compare
leeclemnet
marked this pull request as ready for review
September 29, 2026 20:55
leeclemnet
force-pushed
the
feat/cpu-runtimes
branch
from
September 29, 2026 21:30
d955761 to
2fd7dbd
Compare
2 of 7 tasks
leeclemnet
force-pushed
the
feat/cpu-runtimes
branch
2 times, most recently
from
September 30, 2026 16:53
65ed1a7 to
b40aaf6
Compare
Add OpenVINORuntime on the Runtime contract from the previous PR. - OpenVINORuntime times only InferRequest.infer(). It reads .onnx, .xml, or a directory with one .xml, and sets INFERENCE_PRECISION_HINT from the row precision. - OpenVINO compiles the existing ONNX files on the host, as TensorRT does. rfdetr, yolov8, yolov11 and yolo26 get OpenVINO fp32 and fp16 rows after the old rows. - UnavailableOnHost: a runtime raises it at load when this host cannot run the row as requested (OpenVINO fp16 on a CPU without native fp16), and the runner skips the row. - Extra openvino; nvidia now includes it. CI installs it. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Add large, xlarge and xxlarge to benchmark_rfdetr.py, with the same rows as nano, small and medium: TensorRT fp32 and fp16, ONNX Runtime CPU, and OpenVINO fp32 and fp16. The old rows keep their order. rf-detr-large.onnx in the bucket is the deprecated large model (RFDETRLargeDeprecated). The rows use rf-detr-large-new.onnx, the current RFDETRLarge. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
leeclemnet
force-pushed
the
feat/cpu-runtimes
branch
from
September 30, 2026 19:35
b40aaf6 to
fd3ea3e
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What does this PR do?
Native OpenVINO rows, and all RF-DETR detection sizes
What it does: Adds
OpenVINORuntimeon theRuntimecontract from #15. OpenVINO compiles the existing.onnxfiles on the host, as TensorRT does, sorfdetr,yolov8,yolov11andyolo26get OpenVINO fp32 and fp16 rows with no new artifacts.benchmark_rfdetr.pyalso gets the large, xlarge and xxlarge sizes.Why: Part of the work toward benchmarks of every export format that rfdetr and Ultralytics share. Stacked on #15. LiteRT and ExecuTorch are in their own PRs, because they need exported artifacts.
Change map
sab/runtimes/openvino.py) — times onlyInferRequest.infer(). It reads.onnx,.xml, or a directory with one.xml, and setsINFERENCE_PRECISION_HINTfrom the row precision. Inputs are cast to the model's input dtypes before the timed call, and outputs are copied after it.input_shapereshapes a dynamic image input to a static shape before compilation. The OpenVINO IR from the rf-detr export has the input[?,?,?,?]. A dynamic image input withoutinput_shapefails at load.sab/runtimes/base.py,sab/runner.py) —UnavailableOnHost: a runtime raises it at load when the host cannot run the row as requested, and the runner skips the row. The first case is OpenVINO fp16 on a CPU with no native fp16, which compiles in f32.sab/models/benchmark_{rfdetr,yolov8,yolov11,yolo26}.py) — OpenVINO fp32 and fp16 for each existing.onnxfile, after the old rows. RF-DETR large, xlarge and xxlarge get the same rows as the other sizes.pyproject.toml,uv.lock,.github/workflows/tests.yml,README.md) — extraopenvino.nvidianow includes it.Behavior changes
openvinoextra (includingnvidia) → also runs 2 OpenVINO rows for each ONNX file. Use--runtimesto leave them out.benchmark_rfdetr.py→ 30 rows (was 9): 6 sizes × (TensorRT fp32, TensorRT fp16, ORT CPU, OpenVINO fp32, OpenVINO fp16).Watch out for
sab/models/benchmark_rfdetr.py:56—largeusesrf-detr-large-new.onnx.rf-detr-large.onnxin the bucket is the deprecatedRFDETRLargeDeprecated.sab/runtimes/openvino.py:89— the fp16 check compares the compiled precision with the request. On x86 CPUs with no native fp16, every OpenVINO fp16 row skips.sab/runtimes/openvino.py:126— outputs keep the dtype of the model (for example int64 labels), as the ONNX Runtime rows do.pyproject.toml:35—nvidianow installs OpenVINO, so an NVIDIA host runs the OpenVINO CPU rows by default.Related Issue(s): n/a
Type of Change
Testing
RF-DETR rows use the official exports in
gs://rfdetr/export-2026-09-30: ORT reads the.onnx, and OpenVINO reads the OpenVINO IR (.xml). YOLO26 rows use the bucket.onnxfor both runtimes.CPU: T4 VM (Haswell, 8 cores × 2 HT, fp32)
rfdetr-nanorfdetr-smallrfdetr-mediumrfdetr-largerfdetr-xlargerfdetr-2xlargeyolo26n.onnxyolo26s.onnxyolo26m.onnxyolo26l.onnxyolo26x.onnxBoth runtimes use all 16 logical CPUs. By default, the OpenVINO
LATENCYhint uses only the 8 physical cores.ENABLE_HYPER_THREADING=Trueremoves this limit.CPU: Apple M4 Max (12 performance + 4 efficiency cores, on battery)
rfdetr-nanorfdetr-smallrfdetr-mediumrfdetr-largerfdetr-xlargerfdetr-2xlargeyolo26n.onnxyolo26s.onnxyolo26m.onnxyolo26l.onnxyolo26x.onnxfairor higher).Checklist
Additional Context
🤖 Generated with Claude Code