custom-operator

Here are 2 public repositories matching this topic...

triple-mu / fast-ulysses

Ulysses sequence-parallel all-to-all as a torch custom op, moved by the GPU copy engines into torch symmetric memory. Zero SM usage; 1.66-2.17x over torch.distributed on NVLink.

cuda ulysses pytorch nvlink diffusion-models collective-communication sequence-parallelism custom-operator

Updated Aug 25, 2026
Python

So-cean / mamba-ascendc

Star

Native AscendC Mamba2 selective scan / SSD forward-backward custom operator for Huawei Ascend 910B3 and 950PR, with CANN, torch_npu, A100 benchmarks and msprof profiling.

pytorch ssd mamba cann performance-optimization state-space-models ascend-npu selective-scan ai-infrastructure ascendc kernel-optimization mamba2 huawei-ascend torch-npu custom-operator ascend-910b triton-ascend ascend-950

Updated Aug 12, 2026
Python

Add this topic to your repo

To associate your repository with the custom-operator topic, visit your repo's landing page and select "manage topics."

Learn more