Based on the error message it seems that an internal check failed for the sampleMultinomialOnce kernel.
However, I cannot find the THCNumerics<T>::ge(val, zero) operationand guess that you might be hitting one of these checks 1. Which PyTorch version are you using and could you update to the latest release in case you are using an older one?