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

[PyTorch] Use uint32_t for ProcessedNode::num_outputs (#121335)

Browse Source 
We already use uint32_t for indexing, and the notion of a single graph node with more than four billion outputs stretches credulity.

Differential Revision: [D54598821](https://our.internmc.facebook.com/intern/diff/D54598821/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/121335
Approved by: https://github.com/Skylion007
This commit is contained in:
Scott Wolchok
2024-03-06 13:18:25 -08:00
committed by PyTorch MergeBot
parent ea8f6e2e54
commit 4c58f2b675
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/csrc/jit/runtime/static/impl.h
View File

@ -938,9 +938,9 @@ class TORCH_API ProcessedNode {
return values_[outputs_offset_ + i];
}
size_t num_outputs() const {
uint32_t num_outputs() const {
DCHECK(fn_ != nullptr);
return fn_->num_outputs();
return static_cast<uint32_t>(fn_->num_outputs());
}
C10_NODISCARD c10::ArrayRef<const IValue> outputs() const {