1 files changed, 4 insertions, 0 deletions
diff --git a/src/ProtoGen/SourceGeneratorBase.cs b/src/ProtoGen/SourceGeneratorBase.cs
index 5fa420fc..a7dc2f93 100644
--- a/src/ProtoGen/SourceGeneratorBase.cs
+++ b/src/ProtoGen/SourceGeneratorBase.cs
@@ -43,6 +43,7 @@ namespace Google.ProtocolBuffers.ProtoGen {
     protected readonly bool OptimizeSpeed;
     protected readonly bool OptimizeSize;
     protected readonly bool UseLiteRuntime;
+    protected readonly string RuntimeSuffix;
 
     protected SourceGeneratorBase(T descriptor) {
       this.descriptor = descriptor;
@@ -50,6 +51,9 @@ namespace Google.ProtocolBuffers.ProtoGen {
       OptimizeSize = descriptor.File.Options.OptimizeFor == Google.ProtocolBuffers.DescriptorProtos.FileOptions.Types.OptimizeMode.CODE_SIZE;
       OptimizeSpeed = descriptor.File.Options.OptimizeFor == Google.ProtocolBuffers.DescriptorProtos.FileOptions.Types.OptimizeMode.SPEED;
       UseLiteRuntime = descriptor.File.Options.OptimizeFor == Google.ProtocolBuffers.DescriptorProtos.FileOptions.Types.OptimizeMode.LITE_RUNTIME;
+      //Lite runtime uses OptimizeSpeed code branches
+      OptimizeSpeed |= UseLiteRuntime;
+      RuntimeSuffix = UseLiteRuntime ? "Lite" : "";
     }
 
     protected T Descriptor {