medmekk HF Staff commited on
Commit
ab82ee4
·
verified ·
1 Parent(s): 30486b7

Upload torch-ext/torch_binding.cpp with huggingface_hub

Browse files
Files changed (1) hide show
  1. torch-ext/torch_binding.cpp +1 -1
torch-ext/torch_binding.cpp CHANGED
@@ -4,7 +4,7 @@
4
  #include "torch_binding.h"
5
 
6
  TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
7
- ops.def("rmsnorm_metal_forward(Tensor! input, Tensor weight, Tensor! output) -> ()");
8
 
9
  ops.impl("rmsnorm_metal_forward", torch::kMPS, rmsnorm_metal_forward);
10
  }
 
4
  #include "torch_binding.h"
5
 
6
  TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
7
+ ops.def("rmsnorm_metal_forward(Tensor! input, Tensor! weight, Tensor! output) -> ()");
8
 
9
  ops.impl("rmsnorm_metal_forward", torch::kMPS, rmsnorm_metal_forward);
10
  }