Skip to content

Commit

Permalink
chore: make engine caching opt-in feature (#3152)
Browse files Browse the repository at this point in the history
  • Loading branch information
peri044 authored Sep 11, 2024
1 parent a6d37ee commit a689a3b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions py/torch_tensorrt/dynamo/_defaults.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,8 +35,8 @@
tempfile.gettempdir(), "torch_tensorrt_engine_cache", "timing_cache.bin"
)
LAZY_ENGINE_INIT = False
CACHE_BUILT_ENGINES = True
REUSE_CACHED_ENGINES = True
CACHE_BUILT_ENGINES = False
REUSE_CACHED_ENGINES = False
ENGINE_CACHE_DIR = os.path.join(tempfile.gettempdir(), "torch_tensorrt_engine_cache")
ENGINE_CACHE_SIZE = 1073741824
CUSTOM_ENGINE_CACHE = None
Expand Down

0 comments on commit a689a3b

Please sign in to comment.