[fx] fixed unit tests for torch 1.12 (#1327)

2025-09-04 10:34:41 +00:00 · 2022-07-15 18:22:15 +08:00
parent d49708ae43
commit b2475d8c5c
17 changed files with 2 additions and 18 deletions
--- a/colossalai/fx/tracer/_tracer_utils.py
+++ b/colossalai/fx/tracer/_tracer_utils.py
@@ -22,7 +22,6 @@ def extract_meta(*args, **kwargs):
        if isinstance(val, MetaDeviceAttribute):
            return 'meta'
        elif isinstance(val, ColoProxy):
-            assert val.meta_data is not None
            return val.meta_data
        return val