ocannl icon indicating copy to clipboard operation
ocannl copied to clipboard

Move the `%cd` syntax to the arrayjit library / package

Open lukstafi opened this issue 1 year ago • 1 comments

This would go a long way in making arrayjit useful independently of OCANNL. Some complications might come from interactions with shape inference. But the biggest obstacle is that %cd also interoperates with tensors via including the forward code. If we just move it and don't worry about features unsupported outside of OCANNL, should still work.

lukstafi avatar Apr 01 '24 11:04 lukstafi

We should move shape inference to arrayjit, and add to arrayjit a "tensor light" variant of Tensor that does rootness tracking and shape inference handling, without backpropagation.

lukstafi avatar Apr 24 '24 10:04 lukstafi

Don't agree anymore with the previous comment about rootness tracking. Regarding shape inference, it's tricky. Regarding %cd, it wouldn't work without shape inference. Altogether this needs to be driven by some specific motivation.

lukstafi avatar Jul 24 '24 07:07 lukstafi