Only print out diagnostic messages if an environment variable is set

This prevents crashes in test_core.py due to too many diagnostics emitted in llvm/llvm-project#78228
It should also speed up compile times, as we can use multithreading, and avoid handling diagnostic messages.

PiperOrigin-RevId: 607293980

Author: gflegar

jax_triton/triton_lib.py

@@ -153,13 +153,6 @@ def aval_size_bytes(aval):
   return np.dtype(aval.dtype).itemsize * aval.size
 
 
-# Triton unconditionally enables debug mode in all of its stages. Ideally,
-# this should be controlled by cb.CUDAOptions.debug, but for simplicity we
-# disable debug mode entirely until https://github.com/openai/triton/pull/3127
-# is merged.
-tl_ir.pass_manager.enable_debug = lambda self: None
-
-
 def compile_ttir_to_ptx_inplace(
     ttir,
     cuda_backend: cb.CUDABackend,