coreylowman / cudarc

Safe rust wrapper around CUDA toolkit

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Support for F16 and BF16 with rand_normal?

EricLBuehler opened this issue · comments

Hello everbody,

The Candle implementation of Cuda rand_normal explicitly excludes F16 and BF16 types: here

Is there a reason with cudarc that this is done? Looking at the docs, it seems like there are no limitations on what types, and the fill_with_normal method accepts any T.

Would I be correct to infer that it would be valid to remove these checks altogether? Thank you!