[Feature](mlu-ops): add mluAdamW.

Cambricon · Dec 23, 2024 · a909ce7 · a909ce7
1 parent c952491
commit a909ce7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bangc_kernels.h b/bangc_kernels.h
@@ -23,11 +23,11 @@
 #ifndef BANGC_KERNELS_H_
 #define BANGC_KERNELS_H_
 
-#ifndef NAMESPACE_BANGC_KERNELS_GEGIN
-#define NAMESPACE_BANGC_KERNELS_GEGIN namespace bangc_kernels {
+#ifndef NAMESPACE_BANGC_KERNELS_BEGIN
+#define NAMESPACE_BANGC_KERNELS_BEGIN namespace bangc_kernels {
 #endif
 
-NAMESPACE_BANGC_KERNELS_GEGIN
+NAMESPACE_BANGC_KERNELS_BEGIN
 
 #ifndef BANGC_KERNELS_WIN_API
 #ifdef _WIN32

diff --git a/kernels/adam_w/adam_w_union1.mlu b/kernels/adam_w/adam_w_union1.mlu
@@ -237,7 +237,7 @@ mluOpStatus_t MLUOP_WIN_API KernelApplyAdamW(
   return MLUOP_STATUS_SUCCESS;
 }
 
-NAMESPACE_BANGC_KERNELS_GEGIN
+NAMESPACE_BANGC_KERNELS_BEGIN
 
 template <typename T>
 bangcKernelsStatus_t BANGC_KERNELS_WIN_API