triton code add notations for rejection_sampler.py

Signed-off-by: yuxingcyx <[email protected]>

Author: yuxingcyx

vllm_ascend/sample/rejection_sampler.py

@@ -156,7 +156,7 @@ def rejection_sample(
             BLOCK_SIZE = 2
             grid = triton.cdiv(n, BLOCK_SIZE)
             if n >= 40:
-                grid = 40
+                grid = 40 # Empirically tuned value
                 BLOCK_SIZE = triton.next_power_of_2(n // grid)
 
             if min(num_draft_tokens) == 1 and max(