Changes for review comments incorporated

Signed-off-by: Anurag Dixit <[email protected]>

Author: Anurag Dixit

core/compiler.cpp

@@ -1,8 +1,8 @@
+#include <cuda_runtime.h>
 #include <iostream>
 #include <memory>
 #include <sstream>
 #include <vector>
-#include <cuda_runtime.h>
 
 #include <cuda_runtime.h>
 #include "NvInfer.h"
@@ -48,7 +48,6 @@ void AddEngineToGraph(
     torch::jit::script::Module mod,
     std::shared_ptr<torch::jit::Graph>& g,
     std::string& serialized_engine) {
-
   runtime::CudaDevice device;
 
   // Read current CUDA device properties
@@ -184,7 +183,7 @@ torch::jit::script::Module CompileGraph(const torch::jit::script::Module& mod, C
 }
 
 void set_device(const int gpu_id) {
-    TRTORCH_CHECK((cudaSetDevice(gpu_id) == cudaSuccess), "Unable to set CUDA device: " << gpu_id);
+  TRTORCH_CHECK((cudaSetDevice(gpu_id) == cudaSuccess), "Unable to set CUDA device: " << gpu_id);
 }
 
 } // namespace core

core/compiler.h

@@ -21,7 +21,5 @@ torch::jit::script::Module CompileGraph(const torch::jit::script::Module& module
 
 void set_device(const int gpu_id);
 
-void set_device(const int gpu_id);
-
 } // namespace core
 } // namespace trtorch