-
Notifications
You must be signed in to change notification settings - Fork 360
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
feat: C++ runtime on Windows #2806
Conversation
ca990f0
to
5aad5b7
Compare
Some observations about the failed tests.
|
68bcec0
to
f4ec270
Compare
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.
This looks great and thank you very much for the contribution. Added a few questions/comments. The comments on the test cases + added fixes look good and switching np.dtype("i")
to np.int32
for that case is fine to include in the converter.
1d827f8
to
1723246
Compare
d52ac0f
to
6b7a9e0
Compare
6b7a9e0
to
8915f84
Compare
No ciflow labels are configured for this repo. |
8915f84
to
361e937
Compare
a065cb3
to
137dfba
Compare
Hi @HolyWu - I wanted to share a few notes on the C++ runtime on Windows after some local testing:
Based on the GHA-installed package, I ran the CI-failing tests, for which I have the following results:
|
Was added in pytorch#2750
Linux workflow also does this way so the cuda version of TensorRT doesn't necessarily have to match up PyTorch's cuda version.
They don't always get installed automatically on CI and hence lead to "Can not find nvinfer" error.
Installing torchvision with legacy resolver could cause the installed version of torch conflict with the required version from torchvision, leading to errors like `RuntimeError: operator torchvision::nms does not exist` or `AttributeError: partially initialized module 'torchvision' has no attribute 'extension'`
533bacc
to
11d0293
Compare
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, Thanks for all the hard work!
Description
Fix and enable C++ runtime on Windows.
Fixes #2247
Fixes #2371
Fixes #2484
Type of change
Checklist: