Probability / logit of output tokens

[Simratpreet]

I am trying to log the probability/logit of output token in top K sampling kernel. This is the place where I am trying to log.

My understanding is that outputLogProbs should contain the probability but it is always 0. Also my understanding is sSum variable should contain sum of all logits, but it is always equal to expLogit.

here is actual output for one of the selected tokens:
outputLogProbs[curSeqLen * maxBatchSize + batchSlot]= 0.000000
sSum: 0.000201, logSum: -8.511453, logprob: -8.511453, explogit: 0.000201

In summary:

1. Is this the right place to get the final logit of the token in Top K sampling mode
2. Why would the outputLogprobs be zero and sSum match expLogit? Is this expected?