Skip to content

TensorRT support #2

Closed
Closed

Description

@zhiweit

Before opening: docs/README.md lists every page; docs/pitfalls.md covers the mistakes that produce plausible wrong numbers.

Question
Hi thank you for the contribution and hard work. Will you be willing to provid support for tensorrt inference?
What you tried

Activity

  1. Maelic commented on Sep 22, 2026

    @Maelic
    Owner

    Hi, thank you for your question. Yes I am planning to add TensorRT support, but I can't give a timeline for now, stay tuned!

  2. Maelic commented on Sep 22, 2026

    @Maelic
    Owner

    TensorRT support is implemented in draft PR #4: #4.

    It adds a local ONNX-to-TensorRT engine builder, a standalone relation-head API, and --backend tensorrt for the demo. Dynamic predicate vocabulary, calibrated scores and spatial/semantic graph decoding are preserved.

    The initial path is TensorRT 10, FP32 with TF32 disabled. On an RTX 3080 Laptop GPU, the released ViT-S/16+ passed 45 comparisons against ONNX Runtime, including empty/single-region inputs and vocabulary changes: identical valid pair sets, maximum raw-logit difference 0.000201225. All 88 tests pass locally, including native CUDA integration tests.

    The PR includes installation and inference commands in deploy/README.md. Engines are built on the target GPU rather than redistributed. Jetson, the other two towers and reduced precision still need validation. This is ready for review in the draft PR; it is not merged into the release yet.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions