fixes is_cpu_op conversion in constructor

mlcommons · May 13, 2024 · d966ec8 · d966ec8
1 parent 233b475
commit d966ec8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/feeder/et_feeder_node.cpp b/src/feeder/et_feeder_node.cpp
@@ -14,7 +14,7 @@ ETFeederNode::ETFeederNode(std::shared_ptr<ChakraProtoMsg::Node> node) {
     const string& attr_name = attr.name();
 
     if (attr_name == "is_cpu_op") {
-      this->is_cpu_op_ = static_cast<uint32_t>(attr.int32_val());
+      this->is_cpu_op_ = static_cast<bool>(attr.bool_val());
     } else if (attr_name == "num_ops") {
       this->num_ops_ = static_cast<uint64_t>(attr.int64_val());
     } else if (attr_name == "tensor_size") {

diff --git a/src/feeder/et_feeder_node.h b/src/feeder/et_feeder_node.h
@@ -53,7 +53,7 @@ class ETFeederNode {
 
   uint64_t id_;
   std::string name_;
-  uint32_t is_cpu_op_;
+  bool is_cpu_op_;
   uint64_t runtime_;
   uint64_t num_ops_;
   uint32_t tensor_loc_;