| 130 | } |
| 131 | |
| 132 | CUTE_HOST_DEVICE |
| 133 | bool |
| 134 | thread([[maybe_unused]] int tid, [[maybe_unused]] int bid) |
| 135 | { |
| 136 | #if defined(__CUDA_ARCH__) |
| 137 | return (threadIdx.x + threadIdx.y*blockDim.x + threadIdx.z*blockDim.x*blockDim.y == static_cast<unsigned int>(tid)) && block(bid); |
| 138 | #else |
| 139 | return true; |
| 140 | #endif |
| 141 | } |
| 142 | |
| 143 | CUTE_HOST_DEVICE |
| 144 | bool |