-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
[ROCm][TVMC] Add ROCm to the TVMC driver #8896
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Looks like I missed one of my local changes, to add session.rocm call. Will push update to the PR |
Correct. Sorry I missed that. |
I can see the Jenkins error but unclear if/what I need to change @masahi is there something here related to ROCm? |
No this is a very annoying flaky test that has been happening a lot recently. Please try kicking the CI again. |
Thanks, I may need assistance in restarting it. I don't use this jenkins enough to know how to restart. |
You can do rebase and push, this will cause a new CI job. I also look for a way to retrigger the CI myself. cc @junrushao1994 Do you know how to launch a new CI job as an admin (PMC)? |
* main: [UnitTests][Contrib] Enable contrib tensorrt/coreml unit tests (apache#8902) [BUG] DataType Bug In SplitRel (apache#8899) Enable python debug runtime for exported network libraries (apache#8793) Set default value of p in LpPool as 2 (apache#8866) [Community] @Hzfengsy -> Committer (apache#8908) Trivial uTVM -> microTVM "spelling" fix to align with branding. (apache#8905) [Vulkan][Topi] Parametrizing additional topi tests, marking vulkan failures (apache#8904) Move to new style issue template system (apache#8898) [Onnx] Support Negative Log Loss (apache#8872) [ROCm][TVMC] Add ROCm to the TVMC driver (apache#8896) fix error report on Store (apache#8895) [Docker] Re-enabled automatic --tty flag when running bash. (apache#8861)
* Add ROCm to list of RPC clients. * Add ROCm to list of TVMC devices. * Enable ROCm by adding session call.
* Add ROCm to list of RPC clients. * Add ROCm to list of TVMC devices. * Enable ROCm by adding session call.
Enable ROCm as device for TVMC.