unconditionally add cp to device mesh

$ pytest tests/integration/defs/examples/test_ray.py::test_llm_inference_distributed_ray -s -v

Signed-off-by: Balaram Buddharaju <[email protected]>

Author: brb-nv

tensorrt_llm/_torch/device_mesh.py

@@ -130,9 +130,8 @@ def build_mesh(self):
             dims += ["tp"]
             shape += [self.tp_size]
 
-        if self.cp_size > 1:
-            dims += ["cp"]
-            shape += [self.cp_size]
+        dims += ["cp"]
+        shape += [self.cp_size]
 
         cls.device_mesh = init_device_mesh(
             "cuda",