[Feature](mluAdamW): update adam_w_lite

Cambricon · Jan 23, 2025 · b716cb6 · b716cb6
1 parent 72db95b
commit b716cb6
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/scripts/bangc_kernels_path_config/bangc_kernels_path_extend.json b/scripts/bangc_kernels_path_config/bangc_kernels_path_extend.json
@@ -0,0 +1,13 @@
+{
+  "common": [
+
+  ],
+
+  "operators": [
+    {
+      "name": "mluMoeUnpermute",
+      "header": "kernels/moe_unpermute/moe_unpermute.h",
+      "sources": ["kernels/moe_unpermute"]
+    }
+  ]
+}
diff --git a/tools/commitlint.py b/tools/commitlint.py
@@ -94,8 +94,8 @@ def valid_commit_msg(commit_msg, scopes):
 
         msg_scope = res.groupdict()['scope'].lstrip('(').rstrip(')')
         if msg_scope not in scopes:
-            print("\033[0;35m-- msg_scope not match, the msg_scope should be one of api name 'mluOp*' in mlu_op.h\n "
-                  " or one of lite api name 'mlu(?!Op)*' in ./scripts/bangc_kernels_path_config/*.json or 'mlu-ops' \033[0m")
+            print("\033[0;35m-- msg_scope not match, the msg_scope should be one of api name 'mluOpXXX' in mlu_op.h\n "
+                  " or one of lite api name 'mluXXX' in ./scripts/bangc_kernels_path_config/*.json\n or 'mlu-ops' \033[0m")
             print("\033[0;35m-- please input standard format for commit: {[type](scope): <subject> }\033[0m")
             return False
     return True