Skip to content

Commit 5adad06

Browse files
committed
chore: Apply C++ linting
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 03bafc5 commit 5adad06

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

cpp/bin/torchtrtc/main.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,10 @@ int main(int argc, char** argv) {
5353
{"require-full-compilation"});
5454

5555
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"});
6160

6261
args::Flag disable_tf32(
6362
parser, "disable-tf32", "Prevent Float32 layers from using the TF32 data format", {"disable-tf32"});
@@ -196,7 +195,8 @@ int main(int argc, char** argv) {
196195
// Instead of compiling, just embed engine in a PyTorch module
197196
if (embed_engine) {
198197
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); });
200200

201201
torchtrt::Device device;
202202

@@ -225,7 +225,6 @@ int main(int argc, char** argv) {
225225
return 0;
226226
}
227227

228-
229228
std::vector<torchtrt::Input> ranges;
230229
for (const auto spec : args::get(input_shapes)) {
231230
ranges.push_back(torchtrtc::parserutil::parse_input(spec));
@@ -435,7 +434,8 @@ int main(int argc, char** argv) {
435434
}
436435

437436
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)) {
439439
std::ostringstream threshold_ss;
440440
threshold_ss << threshold_val;
441441
torchtrt::logging::log(

0 commit comments

Comments
 (0)