[hotfix] fixed bugs in ShardStrategy and PcieProfiler (#394)

2025-09-23 10:30:03 +00:00 · 2022-03-11 18:12:46 +08:00
parent 1e4bf85cdb
commit 7c079d9c33
2 changed files with 4 additions and 3 deletions
--- a/colossalai/utils/profiler/pcie_profiler.py
+++ b/colossalai/utils/profiler/pcie_profiler.py
@@ -79,7 +79,7 @@ class PcieProfiler(BaseProfiler):
        if self.profiler.enabled:
            events = self.profiler.function_events
            for event in events:
-                if event.name == "aten::_to_copy":
+                if event.name == "aten::copy_":
                    t_shape = event.input_shapes[0]
                    if len(t_shape) == 0 or event.cuda_time_total == 0:
                        continue