TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Results 599 TensorRT issues
Sort by recently updated
recently updated
newest added

![image](https://github.com/pytorch/TensorRT/assets/58938020/86b22080-6dde-492a-a1e8-cda139766b47) When the three ops above can be supported? Very hope about that! The three ops is related to torch.gather, torch.nn.functional.unfold etc

feature request

## Bug Description ## To Reproduce Steps to reproduce the behavior: 1. 2. 3. ## Expected behavior ## Environment > Build information about Torch-TensorRT can be found by turning on...

bug

## Bug Description ### Bug 1 ```python File "/root/.pyenv/versions/3.10.13/lib/python3.10/site-packages/torch_tensorrt/dynamo/conversion/converter_utils.py", line 491, in to_numpy output = value.cpu().detach().contiguous().numpy() RuntimeError: .numpy() is not supported for tensor subclasses. ``` ### Suggested Fix 1 Need...

bug

## ❓ Question Can torch_tensorRT be used in C++ with multiprocessing using fork? ## What you have already tried I have doubts if this library can be used in C++...

question

- Formalize removal of detach operators pre-AOT via lowering pass

feature request

## ❓ Question ## What you have already tried I'm trying to build **Torch-TensorRT version 2.3.0a0**. I successfully built **Torch 2.3.0.dev**. When building Torch-TensorRT, if I comment **http_archive** for **libtorch**...

question

# Description Support `adaptive_avg_pool1d` dynamo_converter. Fixes #2603 ## Type of change - New feature (non-breaking change which adds functionality) # Checklist: - [x] My code follows the style guidelines of...

component: tests
component: conversion
component: converters
component: api [Python]
cla signed
component: dynamo

# Description Support `adaptive_avg_pool2d` and `adaptive_avg_pool3d`. Fixes #2611 #2612 ## Type of change - New feature (non-breaking change which adds functionality) # Checklist: - [x] My code follows the style...

component: tests
component: conversion
component: converters
component: api [Python]
cla signed
component: dynamo

Fixes #2202

component: conversion
component: converters
component: api [Python]
cla signed
component: dynamo

# Description * This PR adds module_call_graph to preserve the output spec obtained from the graph module. * This will let exported program know what the output_spec is during it's...

component: tests
component: api [Python]
cla signed
component: dynamo
2.2.1