[LayerNorm] Rename layernorm.py -> layer_norm.py

Dao-AILab · Jan 5, 2024 · f5b308e · f5b308e
1 parent 665b55e
commit f5b308e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/flash_attn/ops/triton/layernorm.py → flash_attn/ops/triton/layer_norm.py b/flash_attn/ops/triton/layernorm.py → flash_attn/ops/triton/layer_norm.py
diff --git a/tests/ops/triton/test_layer_norm.py b/tests/ops/triton/test_layer_norm.py
@@ -5,7 +5,7 @@
 import torch.nn.functional as F
 from einops import rearrange, repeat
 
-from flash_attn.ops.triton.layernorm import (
+from flash_attn.ops.triton.layer_norm import (
     layer_norm_fn,
     layer_norm_ref,
     rms_norm_ref,