From 5be1abb4c654279762a463a861526ce4e0c48035 Mon Sep 17 00:00:00 2001
From: Deborah Soung
Date: Wed, 3 Mar 2021 16:45:49 -0800
Subject: Fix ProtoBuf conversions for Verification IR (#2100)

---
 src/main/proto/firrtl.proto                 | 1 +
 src/main/scala/firrtl/proto/FromProto.scala | 7 ++++---
 src/main/scala/firrtl/proto/ToProto.scala   | 1 +
 3 files changed, 6 insertions(+), 3 deletions(-)

(limited to 'src/main')

diff --git a/src/main/proto/firrtl.proto b/src/main/proto/firrtl.proto
index e8451d7a..6ce1c108 100644
--- a/src/main/proto/firrtl.proto
+++ b/src/main/proto/firrtl.proto
@@ -276,6 +276,7 @@ message Firrtl {
       IsInvalid is_invalid = 17;
       MemoryPort memory_port = 18;
       Attach attach = 20;
+      Verification verification = 21;
     }
 
     SourceInfo source_info = 19;
diff --git a/src/main/scala/firrtl/proto/FromProto.scala b/src/main/scala/firrtl/proto/FromProto.scala
index cb9b705e..ed641eb2 100644
--- a/src/main/scala/firrtl/proto/FromProto.scala
+++ b/src/main/scala/firrtl/proto/FromProto.scala
@@ -258,9 +258,10 @@ object FromProto {
       case MEMORY_FIELD_NUMBER          => convert(stmt.getMemory, info)
       case IS_INVALID_FIELD_NUMBER =>
         ir.IsInvalid(convert(info), convert(stmt.getIsInvalid.getExpression))
-      case CMEMORY_FIELD_NUMBER     => convert(stmt.getCmemory, info)
-      case MEMORY_PORT_FIELD_NUMBER => convert(stmt.getMemoryPort, info)
-      case ATTACH_FIELD_NUMBER      => convert(stmt.getAttach, info)
+      case CMEMORY_FIELD_NUMBER      => convert(stmt.getCmemory, info)
+      case MEMORY_PORT_FIELD_NUMBER  => convert(stmt.getMemoryPort, info)
+      case ATTACH_FIELD_NUMBER       => convert(stmt.getAttach, info)
+      case VERIFICATION_FIELD_NUMBER => convert(stmt.getVerification, info)
     }
   }
 
diff --git a/src/main/scala/firrtl/proto/ToProto.scala b/src/main/scala/firrtl/proto/ToProto.scala
index 4cdf6b85..f5ade0e3 100644
--- a/src/main/scala/firrtl/proto/ToProto.scala
+++ b/src/main/scala/firrtl/proto/ToProto.scala
@@ -307,6 +307,7 @@ object ToProto {
               .setCond(convert(cond))
               .setEn(convert(en))
               .setMsg(msg.string)
+            sb.setVerification(vb)
           case ir.IsInvalid(_, expr) =>
             val ib = Firrtl.Statement.IsInvalid
               .newBuilder()
-- 
cgit v1.2.3