chore: Add new parititoning headers to py packaging

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

Author: peri044

py/setup.py

@@ -422,6 +422,9 @@ def run(self):
             "include/torch_tensorrt/core/lowering/*.h",
             "include/torch_tensorrt/core/lowering/passes/*.h",
             "include/torch_tensorrt/core/partitioning/*.h",
+            "include/torch_tensorrt/core/partitioning/segmentedblock/*.h",
+            "include/torch_tensorrt/core/partitioning/partitioninginfo/*.h",
+            "include/torch_tensorrt/core/partitioning/partitioningctx/*.h",
             "include/torch_tensorrt/core/plugins/*.h",
             "include/torch_tensorrt/core/plugins/impl/*.h",
             "include/torch_tensorrt/core/runtime/*.h",