Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit 553c18e

Browse files
author
DEKHTIARJonathan
committed
[Benchmarking-Py] Ensuring default max workspace size >= 8GB
1 parent 24bd2ca commit 553c18e

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,4 +145,4 @@ variables.data*
145145
/*.sh
146146

147147
# Benchmark Result Dir
148-
/benchmark_data
148+
/benchmark_data*

tftrt/benchmarking-python/benchmark_args.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,8 @@ def __init__(self):
195195
self._parser.add_argument(
196196
"--max_workspace_size",
197197
type=int,
198-
default=DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES,
198+
# Ensuring default of minimum 8GB
199+
default=max(1 << 33, DEFAULT_TRT_MAX_WORKSPACE_SIZE_BYTES),
199200
help="The maximum GPU temporary memory which the TRT engine can "
200201
"use at execution time."
201202
)

0 commit comments

Comments
 (0)