Fix scale factor on flops profiler (#4500)

microsoft · Oct 11, 2023 · a25a67a · a25a67a
1 parent 6c86ff3
commit a25a67a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepspeed/profiling/flops_profiler/profiler.py b/deepspeed/profiling/flops_profiler/profiler.py
@@ -722,9 +722,9 @@ def _upsample_flops_compute(*args, **kwargs):
 
     flops = input.numel()
     if isinstance(scale_factor, tuple) and len(scale_factor) == len(input):
-        flops * int(_prod(scale_factor))
+        flops *= int(_prod(scale_factor))
     else:
-        flops * scale_factor**len(input)
+        flops *= scale_factor**len(input)
     return flops, 0