Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "3fc97f67095f5c183a9942bfb45679deab08c127"
Commit 4f7761ae authored by rusty1s's avatar rusty1s
Browse files

typos

parent e83a45a3
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
csrc/cuda/nearest_cuda.cu
View file @ 4f7761ae
...@@ -15,7 +15,7 @@ __global__ void nearest_kernel(const scalar_t *x, const scalar_t *y, ...@@ -15,7 +15,7 @@ __global__ void nearest_kernel(const scalar_t *x, const scalar_t *y,
const int64_t n_x = blockIdx.x; const int64_t n_x = blockIdx.x;
int64_t batch_idx; int64_t batch_idx;
for (int64_t b = 0; b < ptr_x.size(0) - 1; b++) for (int64_t b = 0; b < batch_size; b++)
if (ptr_x[b] >= n_x and ptr_x[b + 1] < n_x) if (ptr_x[b] >= n_x and ptr_x[b + 1] < n_x)
batch_idx = b; batch_idx = b;
...@@ -46,7 +46,7 @@ __global__ void nearest_kernel(const scalar_t *x, const scalar_t *y, ...@@ -46,7 +46,7 @@ __global__ void nearest_kernel(const scalar_t *x, const scalar_t *y,
for (int64_t u = 0; (1 << u) < THREADS; u++) { for (int64_t u = 0; (1 << u) < THREADS; u++) {
__syncthreads(); __syncthreads();
if (idx < (THREADS >> (u + 1))) { if (thread_idx < (THREADS >> (u + 1))) {
int64_t idx_1 = (thread_idx * 2) << u; int64_t idx_1 = (thread_idx * 2) << u;
int64_t idx_2 = (thread_idx * 2 + 1) << u; int64_t idx_2 = (thread_idx * 2 + 1) << u;
if (best_dist[idx_1] > best_dist[idx_2]) { if (best_dist[idx_1] > best_dist[idx_2]) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment