The recent API update in sampling.cuh has introduced compatibility issues with the test code in test_sampling.cu. The function signatures for sampling functions (SamplingFromProb, TopKSamplingFromProb ...