Skip to content
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

Python ncnn with vulkan #5020

Merged
merged 64 commits into from
Oct 23, 2023
Merged

Conversation

Hideousmon
Copy link
Contributor

@Hideousmon Hideousmon commented Sep 10, 2023

Enable the usage of Vulkan with Python wheels. Details and troubleshooting can be found at ncnn-gpu-pre-verification.

Hideousmon and others added 30 commits August 23, 2023 10:18
* add option for enabling vulkan in setup.py
* load winograd type options for pybind.
…etup

prepare for building wheels with vulkan
…etup

enable the usage of humbletim/install-vulkan-sdk in ci
…etup

Compile vulkan in ncnn python setup
@Hideousmon Hideousmon changed the title [WIP] Python ncnn with vulkan Python ncnn with vulkan Sep 14, 2023
@Hideousmon Hideousmon changed the title Python ncnn with vulkan [WIP] Python ncnn with vulkan Sep 15, 2023
@Hideousmon
Copy link
Contributor Author

Hideousmon commented Sep 15, 2023

Found that the building process goes well when resetting -G "Visual Studio 16 2019" -A Win32 for Cmake on x86 Windows. Some improvements again. 😅

@Hideousmon Hideousmon changed the title [WIP] Python ncnn with vulkan Python ncnn with vulkan Sep 17, 2023
@Hideousmon
Copy link
Contributor Author

Finally, it works well on all the platforms. ☺️

x86_64 i686 x86 AMD64 ARM64 universal2 aarch64 ppc64le s390x
ubuntu (cp-manylinux) - - - -
ubuntu (cp-musllinux) - - - -
ubuntu (pp) - - - - - -
windows (pp) - - - - - - - -
windows (cp) - - - - - -
macos (pp) - - - - - - - -
macos (cp) - - - - - -

Updated troubleshooting: https://github.com/Hideousmon/ncnn-gpu-pre-verification/blob/main/troubleshooting.md

@Hideousmon
Copy link
Contributor Author

Hideousmon commented Sep 17, 2023

@Hideousmon
Copy link
Contributor Author

@Hideousmon Hideousmon changed the title Python ncnn with vulkan [WIP] Python ncnn with vulkan Oct 20, 2023
@Hideousmon Hideousmon changed the title [WIP] Python ncnn with vulkan Python ncnn with vulkan Oct 22, 2023
@nihui nihui merged commit a6decad into Tencent:master Oct 23, 2023
17 checks passed
@nihui
Copy link
Member

nihui commented Oct 23, 2023

Thanks for your contribution !

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

Successfully merging this pull request may close these issues.

3 participants