Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit f878256

Browse files
committedMay 25, 2025·
Merge branch 'main' into lluo/fp4_try_out
2 parents f921fc1 + bd74296 commit f878256

File tree

167 files changed

+457
-333
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

167 files changed

+457
-333
lines changed
 

‎README.md

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,9 @@ Torch-TensorRT
99
[![cuda](https://img.shields.io/badge/CUDA-12.8-green)](https://developer.nvidia.com/cuda-downloads)
1010
[![trt](https://img.shields.io/badge/TensorRT-10.9.0-green)](https://github.com/nvidia/tensorrt-llm)
1111
[![license](https://img.shields.io/badge/license-BSD--3--Clause-blue)](./LICENSE)
12-
[![linux_nightly](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux.yml)
13-
[![windows_nightly](https://github.com/pytorch/TensorRT/actions/workflows/build-test-windows.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-windows.yml)
12+
[![Linux x86-64 Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-x86_64.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-x86_64.yml)
13+
[![Linux SBSA Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-aarch64.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-aarch64.yml)
14+
[![Windows Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-windows.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-windows.yml)
1415

1516
---
1617
<div align="left">
@@ -105,9 +106,10 @@ auto results = trt_mod.forward({input_tensor});
105106
| Platform | Support |
106107
| ------------------- | ------------------------------------------------ |
107108
| Linux AMD64 / GPU | **Supported** |
109+
| Linux SBSA / GPU | **Supported** |
108110
| Windows / GPU | **Supported (Dynamo only)** |
109-
| Linux aarch64 / GPU | **Native Compilation Supported on JetPack-4.4+ (use v1.0.0 for the time being)** |
110-
| Linux aarch64 / DLA | **Native Compilation Supported on JetPack-4.4+ (use v1.0.0 for the time being)** |
111+
| Linux Jetson / GPU | **Source Compilation Supported on JetPack-4.4+** |
112+
| Linux Jetson / DLA | **Source Compilation Supported on JetPack-4.4+** |
111113
| Linux ppc64le / GPU | Not supported |
112114
113115
> Note: Refer [NVIDIA L4T PyTorch NGC container](https://ngc.nvidia.com/catalog/containers/nvidia:l4t-pytorch) for PyTorch libraries on JetPack.
@@ -116,10 +118,10 @@ auto results = trt_mod.forward({input_tensor});
116118
117119
These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass.
118120
119-
- Bazel 6.3.2
120-
- Libtorch 2.8.0.dev (latest nightly) (built with CUDA 12.8)
121-
- CUDA 12.8
122-
- TensorRT 10.9.0.43
121+
- Bazel 8.1.1
122+
- Libtorch 2.8.0.dev (latest nightly)
123+
- CUDA 12.8 (CUDA 12.6 on Jetson)
124+
- TensorRT 10.9 (TensorRT 10.3 on Jetson)
123125
124126
## Deprecation Policy
125127

‎docs/_cpp_api/classtorch__tensorrt_1_1DataType.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
<meta name="viewport" content="width=device-width, initial-scale=1.0">
1212

13-
<title>Class DataType &mdash; Torch-TensorRT v2.8.0.dev0+bb990fd documentation</title>
13+
<title>Class DataType &mdash; Torch-TensorRT v2.8.0.dev0+5dbf2b9 documentation</title>
1414

1515

1616

@@ -293,7 +293,7 @@
293293

294294

295295
<div class="version">
296-
v2.8.0.dev0+bb990fd
296+
v2.8.0.dev0+5dbf2b9
297297
</div>
298298

299299

0 commit comments

Comments
 (0)
Please sign in to comment.