From b9cccbf9aea1c93687ad02c2e126df9f0aae2d8a Mon Sep 17 00:00:00 2001
From: Siu Kwan Lam <michael.lam.sk@gmail.com>
Date: Thu, 16 Jan 2014 11:22:52 -0600
Subject: [PATCH] Close everything properly

---
 llvm/workaround/avx_support.py | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/llvm/workaround/avx_support.py b/llvm/workaround/avx_support.py
index 0b1c2da..2e9a751 100644
--- a/llvm/workaround/avx_support.py
+++ b/llvm/workaround/avx_support.py
@@ -12,7 +12,11 @@ http://software.intel.com/sites/default/files/m/a/b/3/4/d/41604-319433-012a.pdf
 
 """
 
-import sys, os, subprocess
+import sys
+import os
+import subprocess
+import contextlib
+
 
 def detect_avx_support(option='detect'):
     '''Detect AVX support'''
@@ -36,13 +40,14 @@ def detect_unix_like():
     except IOError:
         return False
 
-    for line in info:
-        if line.lstrip().startswith('flags'):
-            features = line.split()
-            if 'avx' in features and 'xsave' in features:
-                # enable AVX if flags contain AVX
-                return True
-    return False
+    with contextlib.closing(info):
+        for line in info:
+            if line.lstrip().startswith('flags'):
+                features = line.split()
+                if 'avx' in features and 'xsave' in features:
+                    # enable AVX if flags contain AVX
+                    return True
+        return False
 
 def detect_osx_like():
     try:
@@ -50,9 +55,11 @@ def detect_osx_like():
                                 stdout=subprocess.PIPE)
     except OSError:
         return False
-    features = info.stdout.read().decode('UTF8')
-    features = features.split()
-    return 'AVX1.0' in features and 'OSXSAVE' in features and 'XSAVE' in features
+
+    with info:
+        features = info.stdout.read().decode('UTF8')
+        features = features.split()
+        return 'AVX1.0' in features and 'OSXSAVE' in features and 'XSAVE' in features
 
 
 if __name__ == '__main__':