clean up more

ROCm · Oct 26, 2024 · a9c2ab7 · a9c2ab7
1 parent 9411609
commit a9c2ab7
Show file tree

Hide file tree

Showing 7 changed files with 1 addition and 2,272 deletions.
diff --git a/flash_attn/flash_attn_triton_amd/bwd_prefill.py b/flash_attn/flash_attn_triton_amd/bwd_prefill.py
@@ -2,7 +2,6 @@
 import triton
 import triton.language as tl
 from .utils import get_shape_from_layout, get_strides_from_layout, DEBUG
-DEBUG = False
 
 @triton.jit
 def _bwd_preprocess_use_o(

diff --git a/flash_attn/flash_attn_triton_amd/bwd_ref.py b/flash_attn/flash_attn_triton_amd/bwd_ref.py
@@ -1,7 +1,6 @@
 import torch
 import math
 from .utils import DEBUG
-DEBUG = False
 
 def attention_backward_core_ref_impl(
     do, q, k, v, o, softmax_lse, sm_scale, causal, use_exp2

diff --git a/flash_attn/flash_attn_triton_amd/fwd_old.py b/flash_attn/flash_attn_triton_amd/fwd_old.py
diff --git a/flash_attn/flash_attn_triton_amd/fwd_prefill.py b/flash_attn/flash_attn_triton_amd/fwd_prefill.py
@@ -2,7 +2,6 @@
 import triton
 import triton.language as tl
 from .utils import get_shape_from_layout, get_strides_from_layout, is_cdna, is_rdna, DEBUG, AUTOTUNE
-DEBUG = False
 
 @triton.jit
 def cdiv_fn(x, y):

diff --git a/flash_attn/flash_attn_triton_amd/fwd_ref.py b/flash_attn/flash_attn_triton_amd/fwd_ref.py
@@ -1,7 +1,6 @@
 import torch
 import math
 from .utils import DEBUG
-DEBUG = False
 
 def attention_forward_core_ref_impl(q, k, v, sm_scale, causal, use_exp2):
     if DEBUG: