ivy icon indicating copy to clipboard operation
ivy copied to clipboard

Add automatic device shifting decorator and soft device handling functions

Open bipinKrishnan opened this issue 1 year ago • 1 comments

Adds automatic device shifting functionality like in tensorflow and a soft device variable to control this behaviour.

The PR is related to the task here: https://trello.com/c/TDEyHdDe

bipinKrishnan avatar Jun 23 '23 07:06 bipinKrishnan

If you are working on an open task, please edit the PR description to link to the issue you've created.

For more information, please check ToDo List Issues Guide.

Thank you :hugs:

ivy-leaves avatar Jun 23 '23 07:06 ivy-leaves

Hey @bipinKrishnan, could you please pull from master and resolve the conflicts to see if the fix pushed by @MahmoudAshraf97 resolves the paddle gpu issue? Also it would be great if you could run a few gpu tests locally, just to confirm that we aren't breaking anything (given that there aren't any gpu tests on PRs at the moment 😅)? We could get the PR merged today if everything works fine, thanks 🙂

ivy-gardener

vedpatwardhan avatar Jul 06 '23 06:07 vedpatwardhan

ivy-gardener :white_check_mark: Ivy gardener has formatted your code. If changes are requested, don't forget to pull your fork.

vedpatwardhan avatar Jul 06 '23 07:07 vedpatwardhan

Hey @vedpatwardhan, the device shifting is working as expected on gpu and cpu device with the latest changes made by @MahmoudAshraf97 :slightly_smiling_face:

bipinKrishnan avatar Jul 07 '23 07:07 bipinKrishnan

As discussed, I'm working on a separate PR for the device shifting deep dive @vedpatwardhan

bipinKrishnan avatar Jul 11 '23 05:07 bipinKrishnan