Fix use of blockDim in 'square' sample (#2789)

[ROCm/hip commit: 4d2787749c]
This commit is contained in:
Aiden Nibali
2022-07-07 16:33:19 +10:00
gecommit door GitHub
bovenliggende 013a73583d
commit e6ad33ce12
@@ -38,7 +38,7 @@ THE SOFTWARE.
*/
template <typename T>
__global__ void vector_square(T* C_d, const T* A_d, size_t N) {
size_t offset = (blockIdx.x * blockDim_x + threadIdx.x);
size_t offset = (blockIdx.x * blockDim.x + threadIdx.x);
size_t stride = blockDim.x * gridDim.x;
for (size_t i = offset; i < N; i += stride) {