Skip to content

Pull requests: pytorch/TensorRT

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
Assignee
Filter by who’s assigned
Sort

Pull requests list

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
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
upgrade torch from 2.5.0.dev to 2.6.0.dev in main branch cla signed component: api [C++] Issues re: C++ API component: api [Python] Issues re: Python API component: build system Issues re: Build system component: tests Issues re: Tests
#3165 opened Sep 17, 2024 by lanluo-nvidia Loading…
7 tasks
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
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
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…
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
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
fix: set make_refitable to True when engine cache is used cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths
#3131 opened Aug 30, 2024 by zewenli98 Loading…
7 tasks
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
feat: Implement FP32 accumulation for matmul 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: core Issues re: The core compiler 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
#3110 opened Aug 21, 2024 by peri044 Loading…
7 tasks
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
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
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
Test only: Only run certain test cases if the folder has changed cla signed WIP Work is in progress, pull request should not be merged yet
#2945 opened Jun 21, 2024 by lanluo-nvidia Draft
7 tasks
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
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
feat: Add toggle for fallback to Inductor 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
#2301 opened Sep 7, 2023 by gs-olive Draft
7 tasks
examples: Add example usage script for Function/Module-level acceleration cla signed component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths documentation Improvements or additions to documentation Story: Dynamo Compile Improvements Issues relating to improvement of the Dynamo compile path
#2103 opened Jul 12, 2023 by gs-olive Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.