From 60a91d63afab47a172690974c8b566af74072932 Mon Sep 17 00:00:00 2001
From: jsmall-nvidia <jsmall@nvidia.com>
Date: Tue, 9 Oct 2018 15:09:32 -0400
Subject: Added -serial-ir command line option (#664)

* Added -serial-ir option, to make generateIR always serialize in and out before further processing. Testing out serialization, and adding a kind of 'firewall' between compiler front end and backend.

* Reduce peak memory usage, by discarding IR when stored in serialized form.
Typo fix.
---
 source/slang/slang.cpp | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

(limited to 'source/slang/slang.cpp')

diff --git a/source/slang/slang.cpp b/source/slang/slang.cpp
index 7c06a11e7..e3b675cd7 100644
--- a/source/slang/slang.cpp
+++ b/source/slang/slang.cpp
@@ -9,6 +9,8 @@
 #include "syntax-visitors.h"
 #include "../slang/type-layout.h"
 
+#include "ir-serialize.h"
+
 // Used to print exception type names in internal-compiler-error messages
 #include <typeinfo>
 
@@ -484,7 +486,31 @@ void CompileRequest::generateIR()
     // in isolation.
     for( auto& translationUnit : translationUnits )
     {
-        translationUnit->irModule = generateIRForTranslationUnit(translationUnit);
+        if (useSerialIRBottleneck)
+        {              
+            IRSerialData serialData;
+            {
+                /// Generate IR for translation unit
+                RefPtr<IRModule> irModule(generateIRForTranslationUnit(translationUnit));
+
+                // Write IR out to serialData - copying over SourceLoc information directly
+                IRSerialWriter writer;
+                writer.write(irModule, sourceManager, IRSerialWriter::OptionFlag::RawSourceLocation, &serialData);
+            }
+            RefPtr<IRModule> irReadModule;
+            {
+                // Read IR back from serialData
+                IRSerialReader reader;
+                reader.read(serialData, mSession, irReadModule);
+            }
+
+            // Use the serialized irModule
+            translationUnit->irModule = irReadModule;
+        }
+        else
+        {
+            translationUnit->irModule = generateIRForTranslationUnit(translationUnit);
+        }
     }
 }
 
-- 
cgit v1.2.3