From 51200d03606c1b6a9efc3a3df0ac237ae330a2c3 Mon Sep 17 00:00:00 2001
From: Vinay Ramakrishnaiah <Vinay.Ramakrishnaiah@amd.com>
Date: Fri, 6 Sep 2024 18:59:19 -0400
Subject: [PATCH] merging et_feeder_node

---
 src/feeder/et_feeder_node.cpp | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/feeder/et_feeder_node.cpp b/src/feeder/et_feeder_node.cpp
index 4bdfe426..be58a605 100644
--- a/src/feeder/et_feeder_node.cpp
+++ b/src/feeder/et_feeder_node.cpp
@@ -14,24 +14,26 @@ ETFeederNode::ETFeederNode(std::shared_ptr<ChakraProtoMsg::Node> node) {
     const string& attr_name = attr.name();
 
     if (attr_name == "is_cpu_op") {
-      this->is_cpu_op_ = attr.bool_val();
+      this->is_cpu_op_ = static_cast<bool>(attr.bool_val());
     } else if (attr_name == "num_ops") {
-      this->num_ops_ = attr.uint64_val();
+      this->num_ops_ = static_cast<uint64_t>(attr.int64_val());
     } else if (attr_name == "tensor_size") {
       this->tensor_size_ = attr.uint64_val();
     } else if (attr_name == "comm_type") {
       this->comm_type_ =
           static_cast<ChakraProtoMsg::CollectiveCommType>(attr.int64_val());
     } else if (attr_name == "comm_priority") {
-      this->comm_priority_ = attr.uint32_val();
+      this->comm_priority_ = static_cast<uint32_t>(attr.int32_val());
     } else if (attr_name == "comm_size") {
-      this->comm_size_ = attr.uint64_val();
+      this->comm_size_ = attr.int64_val();
     } else if (attr_name == "comm_src") {
-      this->comm_src_ = attr.uint32_val();
+      this->comm_src_ = static_cast<uint32_t>(attr.int32_val());
     } else if (attr_name == "comm_dst") {
-      this->comm_dst_ = attr.uint32_val();
+      this->comm_dst_ = static_cast<uint32_t>(attr.int32_val());
     } else if (attr_name == "comm_tag") {
-      this->comm_tag_ = attr.uint32_val();
+      this->comm_tag_ = static_cast<uint32_t>(attr.int32_val());
+    } else if (attr_name == "pg_name") {
+      this->pg_name_ = static_cast<string>(attr.string_val());
     } else {
       this->other_attrs_.emplace(attr_name, attr);
     }