Skip to content

Commit 03bafc5

Browse files
committed
refactor!(//cpp/bin/torchtrtc): Removing the max-batch-size argument
BREAKING CHANGE: This PR removes `--max-batch-size` from the CLI as it has no real functional effect Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 0745c5f commit 03bafc5

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

cpp/bin/torchtrtc/main.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,6 @@ int main(int argc, char** argv) {
122122
parser, "num_iters", "Number of averaging timing iterations used to select kernels", {"num-avg-timing-iters"});
123123
args::ValueFlag<uint64_t> workspace_size(
124124
parser, "workspace_size", "Maximum size of workspace given to TensorRT", {"workspace-size"});
125-
args::ValueFlag<uint64_t> max_batch_size(
126-
parser, "max_batch_size", "Maximum batch size (must be >= 1 to be set, 0 means not set)", {"max-batch-size"});
127125
args::ValueFlag<double> threshold(
128126
parser,
129127
"threshold",
@@ -365,7 +363,6 @@ int main(int argc, char** argv) {
365363
compile_settings.workspace_size = args::get(workspace_size);
366364
}
367365

368-
369366
if (truncate_long_and_double) {
370367
compile_settings.truncate_long_and_double = true;
371368
}

docsrc/tutorials/torchtrtc.rst

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,12 @@ to standard TorchScript. Load with ``torch.jit.load()`` and run like you would r
1717
.. code-block:: txt
1818
1919
torchtrtc [input_file_path] [output_file_path]
20-
[input_specs...] {OPTIONS}
20+
[input_specs...] {OPTIONS}
2121
22-
torchtrtc is a compiler for TorchScript, it will compile and optimize
23-
TorchScript programs to run on NVIDIA GPUs using TensorRT
22+
torchtrtc is a compiler for TorchScript, it will compile and optimize
23+
TorchScript programs to run on NVIDIA GPUs using TensorRT
2424
25-
OPTIONS:
25+
OPTIONS:
2626
2727
-h, --help Display this help menu
2828
Verbiosity of the compiler

0 commit comments

Comments
 (0)