From 3aabae6c77c88e520262eccfb5b80f37170625fd Mon Sep 17 00:00:00 2001
From: Xiang Li <python3kgae@outlook.com>
Date: Mon, 23 May 2022 11:04:14 -0700
Subject: [PATCH] Fix crash for llvm-dis for windows build.

---
 tools/llvm-dis/llvm-dis.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/llvm-dis/llvm-dis.cpp b/tools/llvm-dis/llvm-dis.cpp
index f6cc4fe567..6e54671002 100644
--- a/tools/llvm-dis/llvm-dis.cpp
+++ b/tools/llvm-dis/llvm-dis.cpp
@@ -144,13 +144,14 @@ int __cdecl main(int argc, char **argv) { // HLSL Change - __cdecl
   // sys::PrintStackTraceOnErrorSignal(); // HLSL Change - disable this
   // PrettyStackTraceProgram X(argc, argv); // HLSL Change - disable this
   // HLSL Change Starts
+  if (llvm::sys::fs::SetupPerThreadFileSystem())
+    return 1;
   llvm::sys::fs::MSFileSystem* msfPtr;
   HRESULT hr;
   if (!SUCCEEDED(hr = CreateMSFileSystemForDisk(&msfPtr)))
     return 1;
   std::unique_ptr<llvm::sys::fs::MSFileSystem> msf(msfPtr);
   llvm::sys::fs::AutoPerThreadSystem pts(msf.get());
-  llvm::STDStreamCloser stdStreamCloser;
   // HLSL Change Ends
 
   LLVMContext &Context = getGlobalContext();