File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -288,9 +288,9 @@ int main(int argc, char** argv) {
288
288
" (Repeatable) Module that should always be run in Pytorch for execution (partial compilation must be enabled)" ,
289
289
{" tem" , " torch-executed-mods" });
290
290
291
- args::ValueFlagList<std::string > min_block_size (
291
+ args::ValueFlag< uint64_t > min_block_size (
292
292
parser,
293
- " torch-executed-mods " ,
293
+ " min-block-size " ,
294
294
" Minimum number of contiguous TensorRT supported ops to compile a subgraph to TensorRT" ,
295
295
{" mbs" , " min-block-size" });
296
296
@@ -337,10 +337,10 @@ int main(int argc, char** argv) {
337
337
338
338
try {
339
339
parser.ParseCLI (argc, argv);
340
- } catch (args::Help) {
340
+ } catch (args::Help const & ) {
341
341
std::cout << parser;
342
342
return 0 ;
343
- } catch (args::ParseError e) {
343
+ } catch (args::ParseError const & e) {
344
344
torchtrt::logging::log (torchtrt::logging::Level::kERROR , e.what ());
345
345
std::cerr << std::endl << parser;
346
346
return 1 ;
@@ -626,6 +626,7 @@ int main(int argc, char** argv) {
626
626
std::ofstream out (real_output_path);
627
627
out << engine;
628
628
out.close ();
629
+ return 0 ;
629
630
} else {
630
631
auto trt_mod = torchtrt::ts::compile (mod, compile_settings);
631
632
You can’t perform that action at this time.
0 commit comments