From 43d430650f4242800e2bb6ec00b7e8882159e306 Mon Sep 17 00:00:00 2001
From: Adam Yang <31109344+adam-yang@users.noreply.github.com>
Date: Wed, 20 Jan 2021 14:12:13 -0800
Subject: [PATCH] Added the missing thread local IMalloc for possible memory
 allocation in some implementations (#3373)

---
 tools/clang/tools/dxcompiler/dxcpdbutils.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/clang/tools/dxcompiler/dxcpdbutils.cpp b/tools/clang/tools/dxcompiler/dxcpdbutils.cpp
index a24306f3c..c61b7a6e2 100644
--- a/tools/clang/tools/dxcompiler/dxcpdbutils.cpp
+++ b/tools/clang/tools/dxcompiler/dxcpdbutils.cpp
@@ -821,6 +821,8 @@ public:
     if (!m_HasVersionInfo)
       return E_FAIL;
 
+    DxcThreadMalloc TM(m_pMalloc);
+
     CComPtr<DxcPdbVersionInfo> result = CreateOnMalloc<DxcPdbVersionInfo>(m_pMalloc);
     if (result == nullptr) {
       return E_OUTOFMEMORY;