Merge pull request #50 from Qiskit-Partners/opmp_flag_version

Check if compiled with omp from about function

mthree/__init__.py

@@ -23,8 +23,10 @@
 
 try:
     from .version import version as __version__
+    from .version import openmp
 except ImportError:
     __version__ = '0.0.0'
+    openmp = False
 
 from .mitigation import M3Mitigation
 
@@ -35,5 +37,7 @@ def about():
     print('='*80)
     print('# Matrix-free Measurement Mitigation (M3) version {}'.format(__version__))
     print('# (C) Copyright IBM 2021.')
-    print('# Paul D. Nation, Hwajung Kang, Neereja Sundaresan, and Jay Gambetta')
+    print('# Paul Nation, Hwajung Kang, Neereja Sundaresan')
+    print('# Jay Gambetta, and Matthew Treinish.')
+    print('# Compiled with OpenMP: {}'.format(openmp))
     print('='*80)

setup.py

@@ -59,11 +59,13 @@
 # Add openmp flags
 OPTIONAL_FLAGS = []
 OPTIONAL_ARGS = []
+WITH_OMP = False
 for _arg in sys.argv:
     if _arg.startswith("--with-"):
         _options = _arg.split("--with-")[1].split(",")
         sys.argv.remove(_arg)
         if "openmp" in _options or os.getenv("MTHREE_OPENMP", False):
+            WITH_OMP = True
             if sys.platform == 'win32':
                 OPTIONAL_FLAGS = ['/openmp']
             else:
@@ -120,12 +122,12 @@ def write_version_py(filename='mthree/version.py'):
 # pylint: disable=missing-module-docstring
 short_version = '%(version)s'
 version = '%(fullversion)s'
-release = %(isrelease)s
+openmp = %(with_omp)s
 """
     a = open(filename, 'w')
     try:
-        a.write(cnt % {'version': VERSION, 'fullversion':
-                       FULLVERSION, 'isrelease': str(ISRELEASED)})
+        a.write(cnt % {'version': VERSION, 'fullversion':FULLVERSION,
+                       'with_omp': str(WITH_OMP)})
     finally:
         a.close()