chore: Fix centralized partititoning

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

core/partitioning/CMakeLists.txt

@@ -4,6 +4,7 @@ add_library(${lib_name} OBJECT)
 set(CXX_SRCS
     "${CMAKE_CURRENT_SOURCE_DIR}/partitioning.cpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/shape_analysis.cpp"
+    "${CMAKE_CURRENT_SOURCE_DIR}/stitching.cpp"
 )
 
 set(HEADER_FILES
@@ -36,4 +37,4 @@ add_subdirectory(partitioningctx)
 add_subdirectory(partitioninginfo)
 add_subdirectory(segmentedblock)
 
-install(FILES ${HEADER_FILES} DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/torch_tensorrt/core/partitioning")
+install(FILES ${HEADER_FILES} DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/torch_tensorrt/core/partitioning")

core/partitioning/shape_analysis.cpp

@@ -33,8 +33,6 @@ std::unordered_map<const torch::jit::Value*, torch::jit::IValue> generateRandomI
 
   for (auto& input : inputs) {
     if (input.first->type()->kind() == torch::jit::TypeKind::ListType) {
-      // create list
-      std::vector<torch::jit::IValue> list;
       c10::TypePtr elementType = c10::TensorType::get();
       auto generic_list = c10::impl::GenericList(elementType);
       for (size_t i = 0; i < input.second.size(); i++) {