-
Couldn't load subscription status.
- Fork 4.3k
vulkan persistent pipeline cache and shader spirv code cache implementation #6248
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
base: master
Are you sure you want to change the base?
vulkan persistent pipeline cache and shader spirv code cache implementation #6248
Conversation
…test in test_pipeline_cache
…he.h/cpp, reformat code
|
The binary size change of libncnn.so (bytes)
|
|
感谢你的工作,请将你在实现中的笔记和心得,遇到的困难和解决方法等,记录成文章,发表在discussion分区,这将作为知识总结 https://github.com/Tencent/ncnn/discussions Thank you for your work. Please record your notes and experience in the implementation, difficulties encountered and solutions, etc. into an article and publish it in the discussion section. This will serve as a knowledge summary. https://github.com/Tencent/ncnn/discussions |
Codecov Report❌ Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## master #6248 +/- ##
==========================================
- Coverage 95.89% 95.85% -0.04%
==========================================
Files 837 836 -1
Lines 264994 265053 +59
==========================================
- Hits 254105 254076 -29
- Misses 10889 10977 +88 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
Add Persistent Vulkan Pipeline Cache and Persistent Shader SPIR-V Cache
Changes
PipelineCache, enabling serialization ofVkPipelineCachedata to disk and reuse across application runs.Design Considerations
Testing
Performance
In a single pipeline creation test, the time taken for creating a pipeline is reduced from 90ms to 0.4ms using the two caches across runs (mocked by creating and destroying GPU repeatedly) on my PC and this is mainly contributed by spirv code cache.
The CPU is AMD Ryzen 7 5800H and the GPU is Nvidia RTX 3060.
Impact
ncnn_versionin spirv cache header.Questions for Review
1. Cross-platform file operations
The project currently lacks a unified cross-platform file API due to using C++11 (like renaming file). I implemented platform-specific file handling within the relevant
pipelinecache.cppfiles.Would it be better to extract these into a cross-platform file module? Any recommendations or existing patterns in the project?
2. SPIR-V cache invalidation strategy
Multiple factors affect the generated SPIR-V, such as shader source,
glslangversion, and howncnnhandles device extensions and options.I combined these into a single
ncnn_versionfield in the spirv code cache header to invalidate caches on changes.Is this approach sufficient? Are there better versioning or change-tracking strategies you recommend?
3. Testing and API exposure for SPIR-V cache
Exposing many interfaces to manipulate the SPIR-V cache risks complexity, yet hiding them makes thorough testing difficult.
What is the best practice here? Controlled internal APIs for testing, or relying on integration tests?
4. Security of SPIR-V cache files
If a cached SPIR-V file is maliciously altered (content and hash), there's a risk of running compromised shaders when loading the cache.
Do you have suggestions on mitigating this risk?