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

Support for ITensorGPU backend #15

Open
ykharkov opened this issue May 9, 2024 · 0 comments
Open

Support for ITensorGPU backend #15

ykharkov opened this issue May 9, 2024 · 0 comments
Labels
enhancement New feature or request good first issue Good for newcomers

Comments

@ykharkov
Copy link
Member

ykharkov commented May 9, 2024

In the current version the BraketAHS.jl package supports only CPU-based computation.

Goal: Add support for the GPU computations by leveraging ITensorsGPU.jl package.

User should have and ability to select the computational backend in mps_runner.jl.
The GPU acceleration should be enabled for the TEBD evolution loop (and for TDVP once it is implemented).

Options that User can choose from: [CPU (default), GPU].
These options are supplied as a command line argument:
--backend CPU [GPU].

@ykharkov ykharkov added the enhancement New feature or request label May 9, 2024
@kshyatt-aws kshyatt-aws added the good first issue Good for newcomers label May 14, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

2 participants