Skip to content

Commit

Permalink
fix: distingush engines based on compilation settings in addition to …
Browse files Browse the repository at this point in the history
…graph structure

Signed-off-by: Naren Dasan <[email protected]>
  • Loading branch information
narendasan committed Sep 11, 2024
1 parent e2ca04c commit c2e0d4c
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions py/torch_tensorrt/dynamo/_engine_cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,9 @@
import pickletools
import shutil
from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional, Tuple, cast
from typing import Any, Dict, List, Optional, Sequence, Tuple, cast

import torch
from sympy.polys.matrices.dense import Sequence
from torch._inductor.codecache import FxGraphCachePickler, sha256_hash
from torch.fx.experimental.proxy_tensor import unset_fake_temporarily
from torch_tensorrt._Input import Input
Expand Down Expand Up @@ -86,7 +85,7 @@ def pack(
serialized_engine: bytes,
input_names: List[str],
output_names: List[str],
input_specs: Tuple[Input],
input_specs: Sequence[Input],
compilation_settings: CompilationSettings,
weight_name_map: Optional[Dict[Any, Any]],
) -> bytes:
Expand Down

0 comments on commit c2e0d4c

Please sign in to comment.