Skip to content

GitLab

Commit f5b308e2 authored by Tri Dao's avatar Tri Dao
Browse files

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

parent 665b55e2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/ops/triton/test_layer_norm.py
View file @ f5b308e2
...@@ -5,7 +5,7 @@ import torch ...@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment