TensorRT

TensorRT copied to clipboard

🐛 [Bug] Encountered bug when using Torch-TensorRT(torch.nn.LSTM)

2

## Bug Description Encountered error as follow when using Torch-TensorRT to convert torch.nn.LSTM in docker image nvcr.io/nvidia/pytorch:23.12-py3 : NotImplementedError: aten::_cudnn_rnn_flatten_weight: attempted to run this operator with Meta tensors, but there...

johnzlli

# Description support converter for torch.log10 ## Type of change - New feature (non-breaking change which adds functionality) # Checklist: - [x] My code follows the style guidelines of this...

bowang007

## TL;DR ## Goal(s) ## Tasks ```[tasklist] ### Tasks ``` ## Additional context

narendasan

# Description - Add clear logging at the beginning and end of each phase of compilation - Reword logging in certain locations for clarity Addresses #2399 ## Type of change...

gs-olive

Implement embedding bag convertor

5

Christina-Young-NVIDIA

# Description Support converter for torch.log2 Fixes # (issue) ## Type of change - New feature (non-breaking change which adds functionality)

bowang007

## Bug Description arange converter fails with ITensor input ``` File "/home/dperi/Downloads/TensorRT/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py", line 367, in call_function return converter(self.ctx, target, args, kwargs, self._cur_node_name) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/dperi/Downloads/TensorRT/py/torch_tensorrt/dynamo/conversion/ops_evaluators.py", line 49, in aten_ops_arange_start_step return...

peri044

Implement a converter for aten.atan2 https://pytorch.org/docs/stable/generated/torch.atan2.html

narendasan

🐛 [Bug] Errors building Torch-TensorRT on Jetson

3

## Bug Description When trying to build Torch-TensorRT on a Jetson following the instructions [here](https://pytorch.org/TensorRT/getting_started/installation.html), I get errors that seem to be related to changes made to `WORSPACE.jp50` in [this...

airalcorn2

Follow-up to #2502

gs-olive