3 people c25093a305 [Kernel] Batch invariant NVFP4 MoE using cutlass (#40372) ...

Signed-off-by: Jakub Zakrzewski <jzakrzewski@nvidia.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

2026-08-05 04:31:32 +00:00

..

attention

[9/n] Migrate attention and cache kernels to torch stable ABI (continued) (#43717)

2026-05-29 04:44:45 +00:00

core

[ROCm][Quark][7/N] Use MXFP4 linear kernel abstraction for emulation backend (#48949)

2026-07-31 10:07:15 -07:00

cpu

[Bugfix][CPU] Fix macOS build: std::sqrt is not constexpr under libc++ (#50915)

2026-08-04 14:56:33 +00:00

cutlass_extensions

New stable abi cleanup (#46656)

2026-07-03 14:02:26 +08:00

libtorch_stable

[Kernel] Batch invariant NVFP4 MoE using cutlass (#40372)

2026-08-05 04:31:32 +00:00

moe

[CPU][BugFix] Multiple fixes to w4a8_int8 CPU MoE path (#46739)

2026-07-06 05:39:20 +00:00

quantization

[Model] Add Kimi K3 support: model files and kernels [1/N] (#50089)

2026-07-29 14:10:58 +08:00

quickreduce

Enable gfx1250 ROCm architecture (#46516)

2026-07-31 01:41:12 +00:00

rocm

Enable gfx1250 ROCm architecture (#46516)

2026-07-31 01:41:12 +00:00

cache.h

[Bugfix] Sparse MLA: enable fp8_ds_mla dense prefill (#48642)

2026-07-16 22:44:04 +00:00

cuda_compat.h

[Bugfix][ROCm] Fix for warp_size uses on host (#21205)

2025-07-24 00:37:19 -07:00

cuda_utils.h

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

cumem_allocator_compat.h

[Core][AMD] Migrate fully transparent sleep mode to ROCm platform (#12695)

2025-11-12 15:24:12 -08:00

cumem_allocator.cpp

[Test][V1] Add sleep/wake correctness regression test for hybrid GDN/… (#44972)

2026-08-02 22:02:42 +00:00

custom_all_gather_reduce_scatter.cuh

[Model] Add Kimi K3 support: model files and kernels [1/N] (#50089)

2026-07-29 14:10:58 +08:00

custom_all_reduce.cuh

[Model] Add Kimi K3 support: model files and kernels [1/N] (#50089)

2026-07-29 14:10:58 +08:00

custom_collective_common.cuh

[Model] Add Kimi K3 support: model files and kernels [1/N] (#50089)

2026-07-29 14:10:58 +08:00

custom_quickreduce.cu

[ROCM] [Communication] Add INT3 quantization method for quickreduce (#45666)

2026-06-25 15:14:15 +00:00

dispatch_utils.h

[Performance] Fused blockwise quant RMS norm (#27883)

2025-12-07 16:38:04 +00:00

flashkda_registration.cpp

[Kimi-K3] Migrate FlashKDA to PyTorch stable ABI (#50818)

2026-08-03 17:18:08 -07:00

fs_io.cpp

[KV-offload][FS] : Batch store/load_block in C (#49152)

2026-07-28 11:21:53 +08:00

ops.h

[Kernel] Implement CUDA kernel for ReLUSquaredActivation (relu^2) (#39058)

2026-07-12 19:18:03 -07:00

qutlass_registration.cpp

[12/n] final _C library kernel migration (#45415)

2026-06-19 06:57:26 -07:00

spinloop.cpp

[BugFix] Fix clang spinloop mwaitx include (#45532)

2026-07-28 19:00:06 +00:00

torch_bindings.cpp

[ROCm] [PyTorch] Move to stable abi since ROCm upgraded to torch 2.11 (#47128)

2026-07-02 18:34:07 +08:00

torch_utils.h

[6/n] Migrate activation kernels, gptq, gguf, non cutlass w8a8 to libtorch stable ABI (continued) (#42663)

2026-05-20 00:18:12 -07:00