Skip to content

Move fx2trt out of PyTorch core #740

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
Closed

Move fx2trt out of PyTorch core #740

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions torchbenchmark/util/fx2trt.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import torch
from torch.fx.experimental.fx2trt import LowerSetting
from torch.fx.experimental.fx2trt.lower import Lowerer
from fx2trt_oss.fx import LowerSetting
from fx2trt_oss.fx.lower import Lowerer

"""
The purpose of this example is to demostrate the onverall flow of lowering a PyTorch model
Expand Down Expand Up @@ -31,7 +31,7 @@ def lower_to_trt(
explicit_batch_dimension: Use explicit batch dimension in TensorRT if set True, otherwise use implicit batch dimension.
fp16_mode: fp16 config given to TRTModule.
enable_fuse: Enable pass fusion during lowering if set to true. l=Lowering will try to find pattern defined
in torch.fx.experimental.fx2trt.passes from original module, and replace with optimized pass before apply lowering.
in fx2trt_oss.fx.passes from original module, and replace with optimized pass before apply lowering.
verbose_log: Enable verbose log for TensorRT if set True.
timing_cache_prefix: Timing cache file name for timing cache used by fx2trt.
save_timing_cache: Update timing cache with current timing cache data if set to True.
Expand Down