Skip to content

[Releay] Fix on_device call for explicit virtual_device#12088

Merged
masahi merged 1 commit intoapache:mainfrom
Deelvin:amalyshe/python_d_vd
Jul 14, 2022
Merged

[Releay] Fix on_device call for explicit virtual_device#12088
masahi merged 1 commit intoapache:mainfrom
Deelvin:amalyshe/python_d_vd

Conversation

@elvin-n
Copy link
Contributor

@elvin-n elvin-n commented Jul 14, 2022

Previously it was expected that python on_device accept device that is wrong, by fact real on_device op expects virtual_device. If we want to pass memory scope, we have to create explicit virtual_device, but is declined by python wrapper of on_device. Fixed this case

@masahi masahi merged commit 2fee86e into apache:main Jul 14, 2022
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
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

Successfully merging this pull request may close these issues.

2 participants