frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[C10][BE] Use range loop (#131922)

Browse Source 
Non-function change that iterates over entries in `getCollectiveTraceJson` and uses `C10_UNUSED` rather than `(void)i;` trick

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131922
Approved by: https://github.com/XilunWu
This commit is contained in:
Nikita Shulga
2024-07-26 21:58:34 +00:00
committed by PyTorch MergeBot
parent f83ef69b84
commit 07389163f0
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/csrc/distributed/c10d/NCCLUtils.hpp
View File

@ -821,15 +821,12 @@ struct NCCLTraceBuffer {
}
std::list<json> getCollectiveTraceJson(bool onlyActive) {
auto result = dump_entries();
std::list<json> entries;
for (auto i : c10::irange(result.size())) {
json j;
auto& e = result.at(i);
for (auto& e : dump_entries()) {
if (onlyActive && e.time_discovered_completed_.has_value()) {
continue;
}
json j;
j[record_id_key_str] = int64_t(e.id_);
j[pg_id_key_str] = int64_t(e.pg_id_);
j[pg_name_key_str] = e.pg_name_;
@ -846,8 +843,7 @@ struct NCCLTraceBuffer {
auto sizes = std::list<std::list<int>>();
for (auto dim : dims) {
auto arg_sizes = std::list<int>();
for (auto i : c10::irange(dim)) {
(void)i;
for (C10_UNUSED auto i : c10::irange(dim)) {
arg_sizes.push_back(*it++);
}
sizes.push_back(arg_sizes);