From 5b28fd3f7c7877fff0f48f717930e266d0f6e254 Mon Sep 17 00:00:00 2001
From: pnunna93 <104791500+pnunna93@users.noreply.github.com>
Date: Wed, 14 Feb 2024 13:44:30 -0600
Subject: [PATCH] Fix race condition in kEstimateQuantiles (#1061)

---
 csrc/kernels.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/csrc/kernels.cu b/csrc/kernels.cu
index df8488389..6d15dbe64 100644
--- a/csrc/kernels.cu
+++ b/csrc/kernels.cu
@@ -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)
           temp_storage.smem_qidx[j] = -1;
 
+      __syncthreads();
+
       if(threadIdx.x < 256)
       {
           float q_interval = (1.0f-(2.0f*offset))/255.0f;