CUDA implementation for Multinomial op #77
Labels
enhancement
New feature or request
gpu
Issues related to GPU execution
help wanted
Extra attention is needed
The Multinomial op currently falls back on the CPU. This issue tracks the future addition of a dedicated CUDA implementation in
multinomial_gpu.cu
.The text was updated successfully, but these errors were encountered: