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

2024-01-05 00:17:18 -08:00 · 2024-01-05 00:17:18 -08:00 · f5b308e258
commit f5b308e258
parent 665b55e2e2
2 changed files with 1 additions and 1 deletions
--- a/flash_attn/ops/triton/layer_norm.py
+++ b/flash_attn/ops/triton/layer_norm.py
--- a/tests/ops/triton/test_layer_norm.py
+++ b/tests/ops/triton/test_layer_norm.py
@ -5,7 +5,7 @@ import torch
 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,