Fix typo, Update muladd.cpp (#98)

0x00-pl · web-flow · commit e9a1c5d4f808 · 2025-07-08T10:41:05.000-04:00
Should be CPU.
diff --git a/extension_cpp/csrc/muladd.cpp b/extension_cpp/csrc/muladd.cpp
@@ -90,7 +90,7 @@ TORCH_LIBRARY(extension_cpp, m) {
   m.def("myadd_out(Tensor a, Tensor b, Tensor(a!) out) -> ()");
 }
 
-// Registers CUDA implementations for mymuladd, mymul, myadd_out
+// Registers CPU implementations for mymuladd, mymul, myadd_out
 TORCH_LIBRARY_IMPL(extension_cpp, CPU, m) {
   m.impl("mymuladd", &mymuladd_cpu);
   m.impl("mymul", &mymul_cpu);

Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,7 @@ TORCH_LIBRARY(extension_cpp, m) {`
`90`	`90`	`m.def("myadd_out(Tensor a, Tensor b, Tensor(a!) out) -> ()");`
`91`	`91`	`}`
`92`	`92`
`93`		`-// Registers CUDA implementations for mymuladd, mymul, myadd_out`
	`93`	`+// Registers CPU implementations for mymuladd, mymul, myadd_out`
`94`	`94`	`TORCH_LIBRARY_IMPL(extension_cpp, CPU, m) {`
`95`	`95`	`m.impl("mymuladd", &mymuladd_cpu);`
`96`	`96`	`m.impl("mymul", &mymul_cpu);`