add amp alias

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

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"""