From 6f5bf99ed32417cccf632eed7abdc1ef05a42148 Mon Sep 17 00:00:00 2001
From: 0x00-pl <0x00.pl@gmail.com>
Date: Wed, 7 Aug 2024 10:24:25 +0800
Subject: [PATCH] Fix typo, Update muladd.cpp

Should be CPU.
---
 extension_cpp/csrc/muladd.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extension_cpp/csrc/muladd.cpp b/extension_cpp/csrc/muladd.cpp
index 73b8f18..1140677 100644
--- a/extension_cpp/csrc/muladd.cpp
+++ b/extension_cpp/csrc/muladd.cpp
@@ -71,7 +71,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);