Skip to content

Commit 43c850a

Browse files
committed
small fix: Remove extraneous argument in compile
1 parent 3390e24 commit 43c850a

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

py/torch_tensorrt/dynamo/_compiler.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from typing import Any, Collection, List, Optional, Sequence, Set, Tuple, Union
66

77
import torch
8+
import torch_tensorrt
89
from torch.export import ExportedProgram
910
from torch.fx.node import Target
1011
from torch_tensorrt import _enums
@@ -66,8 +67,6 @@
6667
to_torch_tensorrt_device,
6768
)
6869

69-
import torch_tensorrt
70-
7170
logger = logging.getLogger(__name__)
7271

7372

@@ -82,7 +81,6 @@ def compile(
8281
engine_capability: EngineCapability = ENGINE_CAPABILITY,
8382
refit: bool = REFIT,
8483
debug: bool = DEBUG,
85-
capability: EngineCapability = EngineCapability.default,
8684
num_avg_timing_iters: int = NUM_AVG_TIMING_ITERS,
8785
workspace_size: int = WORKSPACE_SIZE,
8886
dla_sram_size: int = DLA_SRAM_SIZE,

0 commit comments

Comments
 (0)