@@ -53,11 +53,10 @@ int main(int argc, char** argv) {
53
53
{" require-full-compilation" });
54
54
55
55
args::ValueFlag<std::string> check_method_op_support (
56
- parser,
57
- " check-method-op-support" ,
58
- " Check the support for end to end compilation of a specified method in the TorchScript module" ,
59
- {" supported" , " is-supported" , " check-support" , " check-method-op-support" }
60
- );
56
+ parser,
57
+ " check-method-op-support" ,
58
+ " Check the support for end to end compilation of a specified method in the TorchScript module" ,
59
+ {" supported" , " is-supported" , " check-support" , " check-method-op-support" });
61
60
62
61
args::Flag disable_tf32 (
63
62
parser, " disable-tf32" , " Prevent Float32 layers from using the TF32 data format" , {" disable-tf32" });
@@ -196,7 +195,8 @@ int main(int argc, char** argv) {
196
195
// Instead of compiling, just embed engine in a PyTorch module
197
196
if (embed_engine) {
198
197
auto device_str = args::get (device_type);
199
- std::transform (device_str.begin (), device_str.end (), device_str.begin (), [](unsigned char c) { return std::tolower (c); });
198
+ std::transform (
199
+ device_str.begin (), device_str.end (), device_str.begin (), [](unsigned char c) { return std::tolower (c); });
200
200
201
201
torchtrt::Device device;
202
202
@@ -225,7 +225,6 @@ int main(int argc, char** argv) {
225
225
return 0 ;
226
226
}
227
227
228
-
229
228
std::vector<torchtrt::Input> ranges;
230
229
for (const auto spec : args::get (input_shapes)) {
231
230
ranges.push_back (torchtrtc::parserutil::parse_input (spec));
@@ -435,7 +434,8 @@ int main(int argc, char** argv) {
435
434
}
436
435
437
436
for (size_t i = 0 ; i < trt_results.size (); i++) {
438
- if (!torchtrtc::accuracy::almost_equal (jit_results[i], trt_results[i].reshape_as (jit_results[i]), threshold_val)) {
437
+ if (!torchtrtc::accuracy::almost_equal (
438
+ jit_results[i], trt_results[i].reshape_as (jit_results[i]), threshold_val)) {
439
439
std::ostringstream threshold_ss;
440
440
threshold_ss << threshold_val;
441
441
torchtrt::logging::log (
0 commit comments