Skip to content

Conversation

@vishwakftw
Copy link
Contributor

Changelog:

  • Migrate _dirichlet_grad implementation from TH to ATen
  • Add CUDA support for _dirichlet_grad

Closes #11030.
Closes #15773.

Changelog:
- Migrate _dirichlet_grad implementation from TH to ATen
- Add CUDA support for _dirichlet_grad
@pytorchbot pytorchbot added module: cpu CPU specific problem (e.g., perf, algorithm) module: cuda Related to torch.cuda, and CUDA support in general module: internals Related to internal abstractions in c10 and ATen module: operators labels May 30, 2019
@vishwakftw vishwakftw closed this May 30, 2019
@vishwakftw
Copy link
Contributor Author

vishwakftw commented May 31, 2019

I forced pushed, and hence can’t reopen the PR. Sorry. New PR opened in #21191 .

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

module: cpu CPU specific problem (e.g., perf, algorithm) module: cuda Related to torch.cuda, and CUDA support in general module: internals Related to internal abstractions in c10 and ATen open source

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Need GPU implementation of dirichlet_grad (originally: Reparameterized gradient on GPU for beta / Dirichlet) implement dirichlet / beta GPU grad

3 participants