Adding NUMBA_ENABLE_PROFILING envvar, enabling jit events and sets default value for NUMBA_DEBUGINFO

anton-malakhov · anton-malakhov · commit 7fada39c6496 · 2018-09-13T14:02:07.000-05:00
diff --git a/numba/config.py b/numba/config.py
@@ -302,10 +302,18 @@ def avx_default():
         NUMBA_NUM_THREADS = _readenv("NUMBA_NUM_THREADS", int,
                                      NUMBA_DEFAULT_NUM_THREADS)
 
+        # Profiling support
+
+        # Indicates if profiler is detected. Only VTune can be detected for now
+        RUNNING_UNDER_PROFILER = 'VS_PROFILER' in os.environ
+
+        # Enables jit events in LLVM in order to support profiling of dynamic code
+        ENABLE_PROFILING = _readenv("NUMBA_ENABLE_PROFILING", int, int(RUNNING_UNDER_PROFILER))
+
         # Debug Info
 
         # The default value for the `debug` flag
-        DEBUGINFO_DEFAULT = _readenv("NUMBA_DEBUGINFO", int, 0)
+        DEBUGINFO_DEFAULT = _readenv("NUMBA_DEBUGINFO", int, ENABLE_PROFILING)
         CUDA_DEBUGINFO_DEFAULT = _readenv("NUMBA_CUDA_DEBUGINFO", int, 0)
 
         # Inject the configuration values into the module globals
diff --git a/numba/targets/codegen.py b/numba/targets/codegen.py
@@ -612,6 +612,9 @@ def _init(self, llvm_module):
         tm = target.create_target_machine(**tm_options)
         engine = ll.create_mcjit_compiler(llvm_module, tm)
 
+        if config.ENABLE_PROFILING:
+            engine.enable_jit_events()
+
         self._tm = tm
         self._engine = JitEngine(engine)
         self._target_data = engine.target_data
