Hi @ahatamiz, could you help release chunk reference code for gated-deltanet-2? without any triton kernels. like what your team do here. https://github.com/NVlabs/GatedDeltaNet/blob/main/lit_gpt/gated_delta_rule_ops/chunk.py#L639
Hi @ahatamiz, could you help release chunk reference code for gated-deltanet-2? without any triton kernels.
like what your team do here. https://github.com/NVlabs/GatedDeltaNet/blob/main/lit_gpt/gated_delta_rule_ops/chunk.py#L639