Skip to content

Commit 191ca97

Browse files
committed
cuda_std: fix some comments.
The indices are derived from the dimensions.
1 parent deaa348 commit 191ca97

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

crates/cuda_std/src/thread.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -100,42 +100,42 @@ macro_rules! in_range {
100100
#[gpu_only]
101101
#[inline(always)]
102102
pub fn thread_idx_x() -> usize {
103-
// The range is derived from the `block_idx_x` range.
103+
// The range is derived from the `block_dim_x` range.
104104
in_range!(core::arch::nvptx::_thread_idx_x, 0..1024) as usize
105105
}
106106

107107
#[gpu_only]
108108
#[inline(always)]
109109
pub fn thread_idx_y() -> usize {
110-
// The range is derived from the `block_idx_y` range.
110+
// The range is derived from the `block_dim_y` range.
111111
in_range!(core::arch::nvptx::_thread_idx_y, 0..1024) as usize
112112
}
113113

114114
#[gpu_only]
115115
#[inline(always)]
116116
pub fn thread_idx_z() -> usize {
117-
// The range is derived from the `block_idx_z` range.
117+
// The range is derived from the `block_dim_z` range.
118118
in_range!(core::arch::nvptx::_thread_idx_z, 0..64) as usize
119119
}
120120

121121
#[gpu_only]
122122
#[inline(always)]
123123
pub fn block_idx_x() -> usize {
124-
// The range is derived from the `grid_idx_x` range.
124+
// The range is derived from the `grid_dim_x` range.
125125
in_range!(core::arch::nvptx::_block_idx_x, 0..2147483647) as usize
126126
}
127127

128128
#[gpu_only]
129129
#[inline(always)]
130130
pub fn block_idx_y() -> usize {
131-
// The range is derived from the `grid_idx_y` range.
131+
// The range is derived from the `grid_dim_y` range.
132132
in_range!(core::arch::nvptx::_block_idx_y, 0..65535) as usize
133133
}
134134

135135
#[gpu_only]
136136
#[inline(always)]
137137
pub fn block_idx_z() -> usize {
138-
// The range is derived from the `grid_idx_z` range.
138+
// The range is derived from the `grid_dim_z` range.
139139
in_range!(core::arch::nvptx::_block_idx_z, 0..65535) as usize
140140
}
141141

0 commit comments

Comments
 (0)