[tracer]add PyTorch 2.4 config

alibaba · Sep 11, 2024 · fca16f3 · fca16f3
1 parent 202aca8
commit fca16f3
Show file tree

Hide file tree

Showing 3 changed files with 1,592 additions and 1 deletion.
diff --git a/tinynn/graph/configs/gen_funcs_yml.py b/tinynn/graph/configs/gen_funcs_yml.py
@@ -160,7 +160,7 @@ def get_scope(ns):
 ver = '_'.join(ver.split('.')[:2])
 
 # Stage 7: Functions in new versions may exist in current version
-latest = '2_0'
+latest = '2_4'
 if ver != latest:
     with open(f'torch_func_override_{latest}.yml', 'r') as f:
         d = yaml.load(f, yaml.SafeLoader)