Open niconiconi opened 1 month ago
https://github.com/PolyhedraZK/ExpanderCompilerCollection/blame/260f4919e7049023e26b0b295eba3ace2b232dbd/docs/docs/cuda/cuda_like_frontend.md#L220
Ideally you can use torch style array, where it's actually a flattened array but has all kinds of shapes.
https://github.com/PolyhedraZK/ExpanderCompilerCollection/blame/260f4919e7049023e26b0b295eba3ace2b232dbd/docs/docs/cuda/cuda_like_frontend.md#L216
https://github.com/PolyhedraZK/ExpanderCompilerCollection/blame/260f4919e7049023e26b0b295eba3ace2b232dbd/docs/docs/cuda/cuda_like_frontend.md#L220