From dc7a556cf46a939d9d8249044296b7b38f6d72ca Mon Sep 17 00:00:00 2001
From: Lehua Ding <lehuading@qq.com>
Date: Mon, 17 Feb 2025 16:54:20 +0800
Subject: [PATCH] Fix typo
Rename CUDA to C++ for C++ implementations
---
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 7b3415f..2ea5213 100644
--- 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 C++ implementations for mymuladd, mymul, myadd_out
TORCH_LIBRARY_IMPL(extension_cpp, CPU, m) {
m.impl("mymuladd", &mymuladd_cpu);
m.impl("mymul", &mymul_cpu);