You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
we have pointwise support in ATen now for CUDA. See https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/cuda/TensorCompare.cu as an example. I'd put it in the same directory in Distributions.cu.
[the build system] should pick up any .cu file there.
If you want half (math) precision support in CUDA, this PR pytorch#4322 does that for embedding, i.e. adds AccumulateTypes, defines the common operators on Half so you don't have to write THCNumerics templates everywhere.
The text was updated successfully, but these errors were encountered:
@gchanan suggests:
The text was updated successfully, but these errors were encountered: