-
-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Labels
enhancementNew feature or requestNew feature or requestphase-1Phase 1 of epic implementationPhase 1 of epic implementation
Description
Phase 1: Enhance Kernel Registry for Vectorized Functions
Part of Epic #247
Objective
Extend the kernel registry system to support vectorized execution hints and registration of both scalar and vectorized implementations for dt-robust operators.
Tasks
Enhance Kernel Registry Types
- Add
VectorizedKernelDescriptorstruct to kernel-registry - Add
VectorizedImplfunction type for vectorized execution - Extend registration to track vectorized implementations
Update Registration Macros
- Add
vectorized = trueattribute to#[kernel_fn]macro - Create
#[vectorized_kernel_fn]macro for vectorized implementations - Update macro documentation with examples
Registry API Extensions
- Add
get_vectorized(name: &str)function to kernel-registry - Add
has_vectorized_impl(name: &str)check function - Update
all_names()to indicate vectorized capability
Testing
- Add tests for vectorized registration
- Test macro expansion with vectorized attributes
- Verify registry lookup functions work correctly
Files to Modify
crates/kernel-registry/src/lib.rscrates/kernel-macros/src/lib.rscrates/kernel-registry/src/types.rs(if exists)
Success Criteria
- Kernel functions can register vectorized implementations
- Registry can lookup vectorized variants
- Macros generate correct registration code
- All tests pass
Dependencies
None - this is the foundation for other phases.
Estimated Time: 1 week
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or requestphase-1Phase 1 of epic implementationPhase 1 of epic implementation