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

Replace TORCH_INTERNAL_ASSERT with TORCH_CHECK (#160411)

Browse Source 
As the title stated.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160411
Approved by: https://github.com/ezyang
This commit is contained in:
FFFrog
2025-08-12 20:31:12 +08:00
committed by PyTorch MergeBot
parent e497620260
commit 8eee08d227
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
c10/cuda/CUDAStream.cpp
View File

@ -147,7 +147,7 @@ static inline StreamIdType streamIdType(StreamId s) {
// rightmost bit
int mask_for_type = (1 << kStreamTypeBits) - 1;
auto val = (s >> 1) & mask_for_type;
TORCH_INTERNAL_ASSERT(val || !(s & 1), "invalid StreamId", s);
TORCH_CHECK(val || !(s & 1), "invalid StreamId", s);
return StreamIdType(val);
}
@ -276,7 +276,7 @@ cudaStream_t CUDAStream::stream() const {
StreamIdType st = streamIdType(stream_id);
size_t si = streamIdIndex(stream_id);
if (st.isDefault()) {
TORCH_INTERNAL_ASSERT(
TORCH_CHECK(
si == 0,
"Unrecognized stream ",
stream_,
@ -291,7 +291,7 @@ cudaStream_t CUDAStream::stream() const {
return reinterpret_cast<cudaStream_t>(stream_id);
} else {
auto streamType = st.getStreamType();
TORCH_INTERNAL_ASSERT(
TORCH_CHECK(
streamType >= 1 && streamType <= max_stream_priorities,
"Unrecognized stream ",
stream_,