1 files changed, 31 insertions, 0 deletions
diff --git a/source/slang/slang.cpp b/source/slang/slang.cpp
index da818984b..ae008cebb 100644
--- a/source/slang/slang.cpp
+++ b/source/slang/slang.cpp
@@ -220,6 +220,9 @@ void Session::init()
     m_languagePreludes[Index(SourceLanguage::CUDA)] = get_slang_cuda_prelude();
     m_languagePreludes[Index(SourceLanguage::CPP)] = get_slang_cpp_prelude();
     m_languagePreludes[Index(SourceLanguage::HLSL)] = get_slang_hlsl_prelude();
+
+    if(!spirvCoreGrammarInfo)
+        spirvCoreGrammarInfo = SPIRVCoreGrammarInfo::getEmbeddedVersion();
 }
 
 void Session::_initCodeGenTransitionMap()
@@ -805,6 +808,34 @@ IDownstreamCompiler* Session::getDownstreamCompiler(CodeGenTarget source, CodeGe
     return getOrLoadDownstreamCompiler(compilerType, nullptr);
 }
 
+SLANG_NO_THROW SlangResult SLANG_MCALL Session::setSPIRVCoreGrammar(char const* jsonPath)
+{
+    if(!jsonPath)
+    {
+        spirvCoreGrammarInfo = SPIRVCoreGrammarInfo::getEmbeddedVersion();
+        SLANG_ASSERT(spirvCoreGrammarInfo);
+    }
+    else
+    {
+        SourceManager* sourceManager = getBuiltinSourceManager();
+        SLANG_ASSERT(sourceManager);
+        DiagnosticSink sink(sourceManager, Lexer::sourceLocationLexer);
+
+        String contents;
+        const auto readRes = File::readAllText(jsonPath, contents);
+        if(SLANG_FAILED(readRes))
+        {
+            sink.diagnose(SourceLoc{}, Diagnostics::unableToReadFile, jsonPath);
+            return readRes;
+        }
+        const auto pathInfo = PathInfo::makeFromString(jsonPath);
+        const auto sourceFile = sourceManager->createSourceFileWithString(pathInfo, contents);
+        const auto sourceView = sourceManager->createSourceView(sourceFile, nullptr, SourceLoc());
+        spirvCoreGrammarInfo = SPIRVCoreGrammarInfo::loadFromJSON(*sourceView, sink);
+    }
+    return spirvCoreGrammarInfo ? SLANG_OK : SLANG_FAIL;
+}
+
 Profile getEffectiveProfile(EntryPoint* entryPoint, TargetRequest* target)
 {
     auto entryPointProfile = entryPoint->getProfile();