I've been trying to test some ops at deepspeed/inference/v2/kernels/ragged_ops, like RaggedLogitsGather, BlockedRotaryEmbeddings, and failed to pass unittest at tests/unit/inference/v2/kernels/ragged_ops on branch "inference".
So I'm wondering
- Are these ops supported now?
- Which branch should I test on?