add amp alias

Jittor · Mar 26, 2022 · 65e9ff1 · 65e9ff1
1 parent e13215b
commit 65e9ff1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py
@@ -9,7 +9,7 @@
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
 
-__version__ = '1.3.1.54'
+__version__ = '1.3.1.55'
 from jittor_utils import lock
 with lock.lock_scope():
     ori_int = int

diff --git a/python/jittor/compiler.py b/python/jittor/compiler.py
@@ -231,6 +231,8 @@ def gen_jit_flags():
             alias = []
             if name == "use_cuda":
                 alias = ["use_device", "use_acl"]
+            elif name == "auto_mixed_precision_level":
+                alias = ["amp_level"]
             get_names = ",".join(["__get__"+a for a in [name]+alias])
             set_names = ",".join(["__set__"+a for a in [name]+alias])
             flags_defs.append(f"""