Skip to content
New issue

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

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

Already on GitHub? # to your account

Tensor doesn't work on GPU #23

Open
alirezamahdavi571 opened this issue Dec 28, 2024 · 1 comment
Open

Tensor doesn't work on GPU #23

alirezamahdavi571 opened this issue Dec 28, 2024 · 1 comment

Comments

@alirezamahdavi571
Copy link

First day that I saw your library, I thought that TensorPAC utilized GPU and I was super exited. But when I used it , I discovered that it doesn't use GPU. can you fix it to run TensorPAC on GPU unit?
please add an example of TensorPAC GPU based.
sincerely,
Alireza

@EtienneCmb
Copy link
Owner

Dear @alirezamahdavi571,

Tensorpac is open-source software. If there's a missing feature, feel free to work on it and make a PR. This should be relatively straightforward given NumPy like API like Cupy, Jax.

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

No branches or pull requests

2 participants