Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Port gamma and dirichlet implementations to CUDA #61

Open
fritzo opened this issue Jan 2, 2018 · 3 comments
Open

Port gamma and dirichlet implementations to CUDA #61

fritzo opened this issue Jan 2, 2018 · 3 comments

Comments

@fritzo
Copy link

fritzo commented Jan 2, 2018

@gchanan suggests:

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.

@fritzo fritzo self-assigned this Jan 2, 2018
@fritzo fritzo removed their assignment Jan 2, 2018
@fritzo
Copy link
Author

fritzo commented Jan 2, 2018

@rachtsingh has already started a CUDA implementation of the Gamma sampler.

@rachtsingh
Copy link

Will land either with or just after #58

@rachtsingh
Copy link

Discussing here: pytorch#4839

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants