Mesa (master): nir: add NIR_PRINT environment variable

[Nicolai Hähnle]

Module: Mesa
Branch: master
Commit: 34df9525f607cab0208ad9b7006b7476f0a0fe79
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=34df9525f607cab0208ad9b7006b7476f0a0fe79

Author: Nicolai Hähnle <nicolai.haehnle at amd.com>
Date:   Sat May 20 17:19:06 2017 +0200

nir: add NIR_PRINT environment variable

Reviewed-by: Rob Clark <robdclark at gmail.com>

---

 src/compiler/nir/nir.h | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h
index 1b4e47058d..c41b0dc931 100644
--- a/src/compiler/nir/nir.h
+++ b/src/compiler/nir/nir.h
@@ -2308,11 +2308,22 @@ should_clone_nir(void)
 
    return should_clone;
 }
+
+static inline bool
+should_print_nir(void)
+{
+   static int should_print = -1;
+   if (should_print < 0)
+      should_print = env_var_as_boolean("NIR_PRINT", false);
+
+   return should_print;
+}
 #else
 static inline void nir_validate_shader(nir_shader *shader) { (void) shader; }
 static inline void nir_metadata_set_validation_flag(nir_shader *shader) { (void) shader; }
 static inline void nir_metadata_check_validation_flag(nir_shader *shader) { (void) shader; }
 static inline bool should_clone_nir(void) { return false; }
+static inline bool should_print_nir(void) { return false; }
 #endif /* DEBUG */
 
 #define _PASS(nir, do_pass) do {                                     \
@@ -2327,14 +2338,22 @@ static inline bool should_clone_nir(void) { return false; }
 
 #define NIR_PASS(progress, nir, pass, ...) _PASS(nir,                \
    nir_metadata_set_validation_flag(nir);                            \
+   if (should_print_nir())                                           \
+      printf("%s\n", #pass);                                         \
    if (pass(nir, ##__VA_ARGS__)) {                                   \
       progress = true;                                               \
+      if (should_print_nir())                                        \
+         nir_print_shader(nir, stdout);                              \
       nir_metadata_check_validation_flag(nir);                       \
    }                                                                 \
 )
 
 #define NIR_PASS_V(nir, pass, ...) _PASS(nir,                        \
+   if (should_print_nir())                                           \
+      printf("%s\n", #pass);                                         \
    pass(nir, ##__VA_ARGS__);                                         \
+   if (should_print_nir())                                           \
+      nir_print_shader(nir, stdout);                                 \
 )
 
 void nir_calc_dominance_impl(nir_function_impl *impl);