Upload torch-ext/torch_binding.cpp with huggingface_hub
Browse files
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 |
}
|