Updated what we're writing out

debugging/checkpointing/checkpointing.proto

@@ -7,3 +7,8 @@ message View {
  required int64  size = 2;
  required bytes  data = 3;
 };
+
+message CoreDump {
+  required int64 num_allocations = 1;
+  repeated View views = 2;
+};

debugging/checkpointing/kp_kernel_logger.cpp

@@ -97,22 +97,25 @@ void dump_checkpoint(int signo){
     size_sum += alloc.second.instances.size();
   }
   out << size_sum;
+  kokkos_checkpointing::CoreDump dump;
+  dump.set_num_allocations(size_sum);
   for(auto& variable_handle: allocations){
     auto& alloc_list = variable_handle.second; 
     for(auto& alloc: alloc_list.instances){
-     kokkos_checkpointing::View v;
-      v.set_size(alloc.how_much);
-      v.set_name(alloc.who);
-      v.set_data(alloc.canonical, alloc.how_much);
-      bool success = v.SerializeToOstream(&out);
-      if(!success){
-        std::cout << "Error serializing a View named "<<alloc.who<<std::endl;
-        exit(1);
-      }
+      kokkos_checkpointing::View* v = dump.add_views();
+      v->set_size(alloc.how_much);
+      v->set_name(alloc.who);
+      v->set_data(alloc.canonical, alloc.how_much);
+      //bool success = v.SerializeToOstream(&out);
       //out << alloc.who << " "<< alloc.how_much; 
       //out.write((char*)alloc.canonical,alloc.how_much);
     }
   }
+  bool success = dump.SerializeToOstream(&out);
+  if(!success){
+    std::cout << "Failed to write checkpoint"<<std::endl;
+  }
+
   std::cout <<"Finished writing on rank "<<rank_string<<std::endl;
   out.close();
 }