Skip to content

Pull requests: pytorch/TensorRT

New pull request New
27 Open 1,545 Closed
27 Open 1,545 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add support for JetPack 6.1 build cla signed component: build system Issues re: Build system documentation Improvements or additions to documentation
#3211 opened Oct 5, 2024 by lanluo-nvidia Loading…
7 tasks
1
feat: Colorize logging output cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: torch_compile
#3210 opened Oct 5, 2024 by HolyWu Loading…
2 of 7 tasks
cherry pick fix global partitioner bug #3195 from main to release/2.5 branch cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: tests Issues re: Tests
#3209 opened Oct 4, 2024 by lanluo-nvidia Loading…
7 tasks
@lanluo-nvidia
chore: adding additional logging to the converter registry system 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
#3199 opened Oct 1, 2024 by narendasan Loading…
7 tasks done
1
fix the global partitioner bug cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: tests Issues re: Tests
#3195 opened Sep 29, 2024 by lanluo-nvidia Loading…
7 tasks
@lanluo-nvidia
10
Require full compilation arg cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: torch_compile
#3193 opened Sep 27, 2024 by apbose Loading…
@apbose
2
Fix for warning as default stream was used in enqueueV3 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
#3191 opened Sep 27, 2024 by keehyuna Loading…
4 of 7 tasks
1
@keehyuna
2
chore: example fixes cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: lowering Issues re: The lowering / preprocessing passes component: torch_compile documentation Improvements or additions to documentation
#3176 opened Sep 24, 2024 by peri044 Loading…
7 tasks
1
Fixed batchnorm bug 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
#3170 opened Sep 20, 2024 by cehongwang • Draft
7 tasks
8
feat: Support weight-stripped engine and REFIT_IDENTICAL flag 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 component: runtime component: tests Issues re: Tests
#3167 opened Sep 19, 2024 by zewenli98 Loading…
7 tasks
1
@zewenli98
28
Run test_base_fp8 for compute capability 8.9 or later cla signed component: tests Issues re: Tests
#3164 opened Sep 17, 2024 by HolyWu Loading…
3 of 7 tasks
1
remove torchvision as dependency cla signed component: api [Python] Issues re: Python API component: tests Issues re: Tests WIP Work is in progress, pull request should not be merged yet
#3158 opened Sep 11, 2024 by lanluo-nvidia • Draft
7 tasks
@lanluo-nvidia
6
MODULE.bazel file for NGC docker container cla signed
#3156 opened Sep 11, 2024 by apbose Loading…
1
scatter reduce lowering with include_self=False cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: lowering Issues re: The lowering / preprocessing passes component: tests Issues re: Tests
#3153 opened Sep 11, 2024 by apbose Loading…
3
feat: log_softmax decomposition 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: tests Issues re: Tests
#3137 opened Sep 1, 2024 by HolyWu Loading…
3 of 7 tasks
21
fix: allow dict type module_outputs in infer_module_output_dtypes 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
#3133 opened Aug 30, 2024 by jiwoong-choi Loading…
2 of 7 tasks
4
feat: Support weight streaming cla signed 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
#3111 opened Aug 22, 2024 by keehyuna Loading…
7 tasks
@keehyuna
57
feat: Implement FP32 accumulation for matmul 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: tests Issues re: Tests component: torch_compile documentation Improvements or additions to documentation
#3110 opened Aug 21, 2024 by peri044 Loading…
7 tasks
7
add runtime_platform_compatibility flag 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 component: runtime component: tests Issues re: Tests WIP Work is in progress, pull request should not be merged yet
#3090 opened Aug 15, 2024 by lanluo-nvidia Loading…
7 tasks
@lanluo-nvidia
5
chore: bug fixes 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
#3065 opened Aug 6, 2024 by peri044 Loading…
7 tasks
11
Enabled python runtime saving cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: runtime
#2999 opened Jul 11, 2024 by cehongwang Loading…
7 tasks
5
example: add example of Phi 3 vision model using torch.compile backend cla signed
#2906 opened Jun 11, 2024 by zewenli98 Loading…
7 tasks
@zewenli98
feat: support aten._local_scalar_dense converter Blocked Issue cannot be resolved until a change in a dependency 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
#2752 opened Apr 16, 2024 by chohk88 Loading…
5 of 7 tasks
1
@chohk88
8
None: Replace the Tensor Layer name from "Unnamed Layer #" to its original name 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 component: fx fx
#2384 opened Oct 9, 2023 by wu6u3tw Loading…
7 tasks
4
chore: enabling torchscript core testing in GHA cla signed
#2303 opened Sep 8, 2023 by narendasan Loading…
4
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.