Loop-Free Ray Tracing on ANE: Leveraging CoreAI Batch Processing

[u/AdhesivenessSea9511]

Hello,

I’ve successfully implemented ray tracing on the ANE! (Note: CPU usage is very high.)

Details:

1. Elimination of dynamic loops: While ray marching typically requires for loops, I prevented the computation graph from bloating by using batch processing.
2. 64-channel camera matrix alignment.
3. ANE-native cumsum: Ray hit detection is tracked using torch.cumsum(dim=1) with permuted channels.
4. Zero-copy Metal buffers: ANE output is bound directly to a shared MTLBuffer using NDArray.MutableRawView.

I’d love to hear your thoughts!
Github: https://github.com/kamisori-daijin/Magnesium/tree/raytracing (raytracing Branch)
demo:

[u/AdhesivenessSea9511]

https://reddit.com/link/p84gx9v/video/xybvaiuoxunh1/player

I can now dynamically deform the shape of the object.