node from aten IR should go to the last if/else, which had a bug so that it will not enter the condition and exec that func. tested on 2.51, should work for 2.6 as well

Signed-off-by: cliu-us <[email protected]>

Author: chichun-charlie-liu

fms_mo/fx/utils.py

@@ -360,7 +360,7 @@ def get_org_mod_name_of_fx_node(
                         org_name = v[: -len(suffix)]
                     break
 
-        if org_name is None:
+        if org_name.startswith("Unknown:"):
             org_name = lname_to_org_name(n_fx_org_mod_name)
 
     return org_name

pyproject.toml

@@ -25,7 +25,7 @@ dependencies = [
 "numpy>=1.26.4,<2.3.0",
 "accelerate>=0.20.3,!=0.34,<1.4",
 "transformers>=4.45,<4.49",
-"torch>=2.2.0,<2.5", 
+"torch>=2.2.0,<2.6", 
 "triton>=3.0,<3.2",
 "tqdm>=4.66.2,<5.0",
 "datasets>=3.0.0,<4.0",