burtenshaw HF Staff commited on
Commit
e443917
·
verified ·
1 Parent(s): e0a93ee

Upload torch-ext/torch_binding.h with huggingface_hub

Browse files
Files changed (1) hide show
  1. torch-ext/torch_binding.h +11 -0
torch-ext/torch_binding.h ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+
3
+ #include <torch/torch.h>
4
+
5
+ void rmsnorm(
6
+ torch::Tensor& output,
7
+ const torch::Tensor& input,
8
+ const torch::Tensor& weight,
9
+ float eps
10
+ );
11
+