Implement pytest for 03_all_store/all_store_bench.py with kernel-level testing #131
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR adds a pytest test file for the all-store benchmark that focuses on testing the core kernel functionality rather than the full benchmarking infrastructure.
Implementation
The test follows the same pattern as
test_load_bench.py
by:all_store_kernel
instead of the fullrun_experiment()
functionint8
,float16
,bfloat16
,float32
), buffer sizes, and block sizesWhy Kernel-Level Testing
The original approach of calling
run_experiment()
caused crashes due to GPU synchronization barriers in the CI environment. By testing theall_store_kernel
directly, we:test_load_bench.py
The test ensures the all-store kernel can be invoked with various data types and configurations, providing confidence in the core functionality without requiring the full multi-GPU benchmarking setup.
Fixes #57.
💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.