pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.61k stars 351 forks source link

cherry pick to release/2.5 fix: Fix static arange export 3194 #3207

Closed lanluo-nvidia closed 1 month ago

lanluo-nvidia commented 1 month ago

Description

cherry pick fix: Fix static arange export #3194 from main to release/2.5 branch

Fixes # (issue)

Type of change

Please delete options that are not relevant and/or add your own.

Checklist: