Upload rmsnorm_metal/rmsnorm.mm with huggingface_hub
Browse files- rmsnorm_metal/rmsnorm.mm +1 -1
rmsnorm_metal/rmsnorm.mm
CHANGED
|
@@ -19,7 +19,7 @@ static std::string getModuleDirectory() {
|
|
| 19 |
return ".";
|
| 20 |
}
|
| 21 |
|
| 22 |
-
|
| 23 |
at::Tensor input,
|
| 24 |
at::Tensor weight,
|
| 25 |
at::Tensor output,
|
|
|
|
| 19 |
return ".";
|
| 20 |
}
|
| 21 |
|
| 22 |
+
void rmsnorm_metal_forward(
|
| 23 |
at::Tensor input,
|
| 24 |
at::Tensor weight,
|
| 25 |
at::Tensor output,
|