squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Tri Dao 2024-01-05 00:17:18 -08:00
parent 665b55e2e2
commit f5b308e258
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
flash_attn/ops/triton/layernorm.py → flash_attn/ops/triton/layer_norm.py
View File

2
tests/ops/triton/test_layer_norm.py
View File

@ -5,7 +5,7 @@ import torch
import torch.nn.functional as F import torch.nn.functional as F
from einops import rearrange, repeat 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_fn,
layer_norm_ref, layer_norm_ref,
rms_norm_ref, rms_norm_ref,