From 9ad819a35501c43273e92b6cbc45947a43b25305 Mon Sep 17 00:00:00 2001
From: Prasanth Nunna <Prasanth.Nunna@amd.com>
Date: Tue, 13 Feb 2024 20:49:20 +0000
Subject: [PATCH] Fix race condition in kEstimateQuantiles

---
 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;