Hide navigation sidebar

Hide table of contents sidebar

Skip to content

Toggle site navigation sidebar

FlashInfer 0.2.11.post1 documentation

Toggle table of contents sidebar

FlashInfer 0.2.11.post1 documentation

Get Started

Installation

Tutorials

Attention States and Recursive Attention
KV-Cache Layout in FlashInfer

PyTorch API Reference

FlashInfer Attention Kernels
Toggle navigation of FlashInfer Attention Kernels
flashinfer.gemm
Toggle navigation of flashinfer.gemm
flashinfer.fused_moe
Toggle navigation of flashinfer.fused_moe
flashinfer.cascade
Toggle navigation of flashinfer.cascade
flashinfer.sparse
flashinfer.page
Toggle navigation of flashinfer.page
flashinfer.sampling
Toggle navigation of flashinfer.sampling
flashinfer.logits_processor
Toggle navigation of flashinfer.logits_processor
flashinfer.norm
Toggle navigation of flashinfer.norm
flashinfer.rope
Toggle navigation of flashinfer.rope
flashinfer.activation
Toggle navigation of flashinfer.activation
flashinfer.quantization
Toggle navigation of flashinfer.quantization
- flashinfer.quantization.packbits
- flashinfer.quantization.segment_packbits
flashinfer.green_ctx
Toggle navigation of flashinfer.green_ctx
- flashinfer.green_ctx.split_device_green_ctx
- flashinfer.green_ctx.split_device_green_ctx_by_sm_count
flashinfer.fp4_quantization
Toggle navigation of flashinfer.fp4_quantization
flashinfer.testing
Toggle navigation of flashinfer.testing

Toggle table of contents sidebar

flashinfer.fused_moe.reorder_rows_for_gated_act_gemm¶

flashinfer.fused_moe.reorder_rows_for_gated_act_gemm(x)¶: PyTorch implementation of trt-llm gen reorderRowsForGatedActGemm

flashinfer.fused_moe.cutlass_fused_moe

flashinfer.fused_moe.convert_to_block_layout

Copyright © 2023-2025, FlashInfer Contributors

Made with Sphinx and @pradyunsg's Furo

On this page

flashinfer.fused_moe.reorder_rows_for_gated_act_gemm
- reorder_rows_for_gated_act_gemm()