[BUG FIX] Refactor kernels for improved pointer handling

[LoserCheems]

Summary

• Refactor multiple kernel functions to enhance pointer handling and memory operations, impacting performance and memory efficiency.

Root Cause

• Previous implementations had inefficient pointer arithmetic and memory access patterns, leading to potential performance bottlenecks.

Changes

• Updated kernel functions to use direct pointer initialization and optimized memory load/store operations.

Reproduction

• No specific bug was identified, but performance improvements can be observed through benchmarking the kernel functions.

Tests

• Ran existing tests for all kernel functions; all tests passed successfully.

Compatibility

• No migration concerns or backwards compatibility issues identified.

Checklist

• Linked issue provided
• Adds or updates tests
• Updates docs if needed
• No perf regressions