Skip to content

Commit

Permalink
Temporarily disabled debug mode in ir.pass_manager.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 606958579
  • Loading branch information
superbobry authored and The jax_triton Authors committed Feb 14, 2024
1 parent 6effbb5 commit 3666738
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions jax_triton/triton_lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,6 +153,13 @@ 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,
Expand Down

0 comments on commit 3666738

Please sign in to comment.