We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent af26ee9 commit c6d9a92Copy full SHA for c6d9a92
1 file changed
mlx/backend/rocm/unary.hip
@@ -52,6 +52,7 @@ __global__ void unary_g(
52
53
auto shape_x = shape[ndim - 1];
54
auto stride_x = strides[ndim - 1];
55
+
56
IdxT index_x = blockIdx.x * blockDim.x + threadIdx.x;
57
58
// Compute base offset for this row using elem_to_loc style calculation
0 commit comments