Skip to content

Commit c2e0d4c

Browse files
committed
fix: distingush engines based on compilation settings in addition to graph structure
Signed-off-by: Naren Dasan <[email protected]>
1 parent e2ca04c commit c2e0d4c

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

py/torch_tensorrt/dynamo/_engine_cache.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,9 @@
66
import pickletools
77
import shutil
88
from abc import ABC, abstractmethod
9-
from typing import Any, Dict, List, Optional, Tuple, cast
9+
from typing import Any, Dict, List, Optional, Sequence, Tuple, cast
1010

1111
import torch
12-
from sympy.polys.matrices.dense import Sequence
1312
from torch._inductor.codecache import FxGraphCachePickler, sha256_hash
1413
from torch.fx.experimental.proxy_tensor import unset_fake_temporarily
1514
from torch_tensorrt._Input import Input
@@ -86,7 +85,7 @@ def pack(
8685
serialized_engine: bytes,
8786
input_names: List[str],
8887
output_names: List[str],
89-
input_specs: Tuple[Input],
88+
input_specs: Sequence[Input],
9089
compilation_settings: CompilationSettings,
9190
weight_name_map: Optional[Dict[Any, Any]],
9291
) -> bytes:

0 commit comments

Comments
 (0)