已合并
fix(test): correct expected buffer size calculation in test_hccl_shared_buffer #39515
fix(test): correct expected buffer size calculation in test_hccl_shared_buffer #39515
已合并
共 1 个文件变更+3-3
| @@ -313,7 +313,7 @@ class HcclSharedBufferTest(TestCase): | |||
| 313 | _, _, mem_after = cls.get_memory_info() | 313 | _, _, mem_after = cls.get_memory_info() |
| 314 | diff = (memory_info_before[0] - mem_after[0]) / 1024 / 1024 | 314 | diff = (memory_info_before[0] - mem_after[0]) / 1024 / 1024 |
| 315 | num_different_buffer_group += 1 | 315 | num_different_buffer_group += 1 |
| 316 | - expected = buffer_size * num_different_buffer_group + tol | 316 | + expected = (buffer_size + tol) * num_different_buffer_group |
| 317 | mem_diff["pg1"] = [diff, expected] | 317 | mem_diff["pg1"] = [diff, expected] |
| 318 | 318 | ||
| 319 | # create pg2 with the same buffer | 319 | # create pg2 with the same buffer |
| @@ -325,7 +325,7 @@ class HcclSharedBufferTest(TestCase): | |||
| 325 | # get pg2 mem info | 325 | # get pg2 mem info |
| 326 | _, _, mem_after = cls.get_memory_info() | 326 | _, _, mem_after = cls.get_memory_info() |
| 327 | diff = (memory_info_before[0] - mem_after[0]) / 1024 / 1024 | 327 | diff = (memory_info_before[0] - mem_after[0]) / 1024 / 1024 |
| 328 | - expected = buffer_size * num_different_buffer_group + tol | 328 | + expected = (buffer_size + tol) * num_different_buffer_group |
| 329 | mem_diff["pg2"] = [diff, expected] | 329 | mem_diff["pg2"] = [diff, expected] |
| 330 | 330 | ||
| 331 | # create pg3 with a different buffer | 331 | # create pg3 with a different buffer |
| @@ -338,7 +338,7 @@ class HcclSharedBufferTest(TestCase): | |||
| 338 | _, _, mem_after = cls.get_memory_info() | 338 | _, _, mem_after = cls.get_memory_info() |
| 339 | diff = (memory_info_before[0] - mem_after[0]) / 1024 / 1024 | 339 | diff = (memory_info_before[0] - mem_after[0]) / 1024 / 1024 |
| 340 | num_different_buffer_group += 1 | 340 | num_different_buffer_group += 1 |
| 341 | - expected = buffer_size * num_different_buffer_group + tol | 341 | + expected = (buffer_size + tol) * num_different_buffer_group |
| 342 | mem_diff["pg3"] = [diff, expected] | 342 | mem_diff["pg3"] = [diff, expected] |
| 343 | torch_npu.npu.synchronize() | 343 | torch_npu.npu.synchronize() |
| 344 | 344 | ||