Hugging Face's logo

medmekk
/
rmsnorm

kernel
Model card Files
xet
 Community
1
medmekk commited on
Commit
bb342db
·
verified ·
1 Parent(s): 07fba08

Update torch-ext/rmsnorm/rmsnorm.py

Browse files
Files changed (1) hide show
  1. torch-ext/rmsnorm/rmsnorm.py +0 -1
torch-ext/rmsnorm/rmsnorm.py CHANGED
@@ -10,7 +10,6 @@ import math
10
 
11
  import torch
12
  import torch.nn.functional as F
13
- from torch.amp import custom_fwd, custom_bwd
14
 
15
  import triton
16
  import triton.language as tl
 
10
 
11
  import torch
12
  import torch.nn.functional as F
 
13
 
14
  import triton
15
  import triton.language as tl