Skip to content

Pull requests: pytorch/TensorRT

New pull request New
39 Open 1,885 Closed
39 Open 1,885 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

fix: set example models to eval mode and follow the convention cla signed
#3770 opened Aug 9, 2025 by zewenli98 Loading…
4 of 7 tasks
@zewenli98
Cherry pick jetson enablement from 2.8 release branch to main ciflow/binaries/all Build for all Python Versions cla signed component: api [Python] Issues re: Python API component: build system Issues re: Build system component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: lowering Issues re: The lowering / preprocessing passes documentation Improvements or additions to documentation
#3765 opened Aug 8, 2025 by lanluo-nvidia Loading…
7 tasks
6
Please do not review: perf test rtx only cla signed WIP Work is in progress, pull request should not be merged yet
#3763 opened Aug 7, 2025 by lanluo-nvidia • Draft
7 tasks
Breaking Change: Remove the deprecated int8 calibrator related ciflow/binaries/all Build for all Python Versions cla signed component: api [C++] Issues re: C++ API component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: runtime component: tests Issues re: Tests fx
#3759 opened Aug 7, 2025 by lanluo-nvidia Loading…
7 tasks
19
fix: batch norm issue encountered in RAFT cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: lowering Issues re: The lowering / preprocessing passes
#3758 opened Aug 7, 2025 by zewenli98 Loading…
5 of 7 tasks
1
@zewenli98
4
Revised the lowering pass according to Bo's suggestion cla signed
#3756 opened Aug 7, 2025 by cehongwang Loading…
7 tasks
Add support for TensorRT-RTX ciflow/binaries/all Build for all Python Versions cla signed component: api [C++] Issues re: C++ API component: api [Python] Issues re: Python API component: build system Issues re: Build system component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: evaluators Issues re: Specific op evaluators component: fx component: lowering Issues re: The lowering / preprocessing passes component: partitioning component: runtime component: tests Issues re: Tests documentation Improvements or additions to documentation fx
#3753 opened Aug 6, 2025 by lanluo-nvidia Loading…
7 tasks
15
fix: atan2 strong type support & bug fix for integer dynamic shape cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths
#3751 opened Aug 6, 2025 by chohk88 Loading…
6 of 7 tasks
@chohk88
2
add strong typing fix cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths
#3749 opened Aug 5, 2025 by lanluo-nvidia • Draft
7 tasks
cherry pick 3700 to 2.8 release: Broadcast removal cla signed component: api [C++] Issues re: C++ API component: api [Python] Issues re: Python API component: build system Issues re: Build system component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: fx component: lowering Issues re: The lowering / preprocessing passes component: runtime component: tests Issues re: Tests component: torch_compile documentation Improvements or additions to documentation fx
#3747 opened Aug 5, 2025 by cehongwang Loading…
7 tasks
fix: Inferred dimensions at build time in reshape cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths
#3746 opened Aug 5, 2025 by keehyuna Loading…
4 of 7 tasks
1
@keehyuna
1
add sliding window support for Gemma3 cla signed
#3742 opened Aug 2, 2025 by lanluo-nvidia • Draft
7 tasks
Tentatively eliminate graph break overhead cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: runtime
#3741 opened Aug 1, 2025 by cehongwang • Draft
7 tasks
1
Feat: Pre-quantized LLM model support cla signed
#3740 opened Aug 1, 2025 by keehyuna Loading…
3 of 7 tasks
@keehyuna
feat: Add support for Groot N1.5 model cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: lowering Issues re: The lowering / preprocessing passes component: runtime
#3736 opened Jul 31, 2025 by peri044 • Draft
7 tasks
feat: check the num of ILayer types and compare with ONNX-TRT in converter tests cla signed component: tests Issues re: Tests
#3732 opened Jul 31, 2025 by zewenli98 Loading…
4 of 7 tasks
@zewenli98
Dynamic memory allocation cla signed component: api [Python] Issues re: Python API component: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: runtime
#3727 opened Jul 29, 2025 by cehongwang Loading…
7 tasks
4
fix: prelu perf gap on Unet cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths
#3717 opened Jul 24, 2025 by zewenli98 Loading…
4 of 7 tasks
1
@zewenli98
9
fix: Remove type casting in matmul and add scalar tensor conversion cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths
#3713 opened Jul 24, 2025 by keehyuna Loading…
7 tasks
1
@keehyuna
addressing ngc aarch64 error cla signed
#3705 opened Jul 23, 2025 by apbose • Draft
integrated vlm code for benchmark for Eagle2 cla signed
#3698 opened Jul 21, 2025 by chohk88 Loading…
6 of 7 tasks
@chohk88
20
Index converter dynamic cases fix cla signed component: api [Python] Issues re: Python API component: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: tests Issues re: Tests
#3694 opened Jul 17, 2025 by apbose Loading…
2
Added warnings if the model is in training mode cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: runtime
#3676 opened Jul 10, 2025 by cehongwang Loading…
7 tasks
chore: move external dep installation into a separate script cla signed component: tests Issues re: Tests
#3672 opened Jul 8, 2025 by peri044 Loading…
7 tasks
3
Lluo/lan fix torch compile gpt2 cla signed component: api [Python] Issues re: Python API component: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: runtime WIP Work is in progress, pull request should not be merged yet
#3671 opened Jul 8, 2025 by lanluo-nvidia • Draft
7 tasks
@lanluo-nvidia
ProTip! Exclude everything labeled bug with -label:bug.