Skip to content

GitLab

"...python/git@developer.sourcefind.cn:change/sglang.git" did not exist on "ecc9f3e47abd8fa1a23020a91b4a50088fd3c060"
Commit f99e7ffb authored by rusty1s's avatar rusty1s
Browse files

fix installation bug + cpu/cuda unequal output bug

parent 234a6ba3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
csrc/cuda/radius_cuda.cu
View file @ f99e7ffb
...@@ -31,7 +31,7 @@ __global__ void radius_kernel(const scalar_t *x, const scalar_t *y, ...@@ -31,7 +31,7 @@ __global__ void radius_kernel(const scalar_t *x, const scalar_t *y,
} }
dist = sqrt(dist); dist = sqrt(dist);
if (dist <= radius) { if (dist < radius) {
row[n_y * max_num_neighbors + count] = n_y; row[n_y * max_num_neighbors + count] = n_y;
col[n_y * max_num_neighbors + count] = n_x; col[n_y * max_num_neighbors + count] = n_x;
count++; count++;
......
setup.py
View file @ f99e7ffb
...@@ -58,7 +58,7 @@ def get_extensions(): ...@@ -58,7 +58,7 @@ def get_extensions():
sources += [path] sources += [path]
path = osp.join(extensions_dir, 'cuda', f'{name}_cuda.cu') path = osp.join(extensions_dir, 'cuda', f'{name}_cuda.cu')
if WITH_CUDA and osp.exists(path): if suffix == 'cuda' and osp.exists(path):
sources += [path] sources += [path]
Extension = CppExtension if suffix == 'cpu' else CUDAExtension Extension = CppExtension if suffix == 'cpu' else CUDAExtension
......
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