Skip to content

GitLab

Unverified Commit 5b28fd3f authored by pnunna93's avatar pnunna93 Committed by GitHub
Browse files

Fix race condition in kEstimateQuantiles (#1061)

parent 344e8516
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
csrc/kernels.cu
View file @ 5b28fd3f
...@@ -654,6 +654,8 @@ __global__ void kEstimateQuantiles(T *__restrict__ const A, float *code, const f ...@@ -654,6 +654,8 @@ __global__ void kEstimateQuantiles(T *__restrict__ const A, float *code, const f
for(int j = threadIdx.x; j < BLOCK_ESTIMATE; j+=blockDim.x) for(int j = threadIdx.x; j < BLOCK_ESTIMATE; j+=blockDim.x)
temp_storage.smem_qidx[j] = -1; temp_storage.smem_qidx[j] = -1;
__syncthreads();
if(threadIdx.x < 256) if(threadIdx.x < 256)
{ {
float q_interval = (1.0f-(2.0f*offset))/255.0f; float q_interval = (1.0f-(2.0f*offset))/255.0f;
......
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