---
id: pr-flashinfer-2587
repo: flashinfer-ai/flashinfer
pr: 2587
title: 'feat: trtllm tinygemm2 in flashinfer as bf16 routergemm'
author: jimmyzho
date: '2026-02-24'
url: https://github.com/flashinfer-ai/flashinfer/pull/2587
source_category: upstream-code
architectures:
- sm100
- sm90
tags:
- gemm
techniques: []
hardware_features: []
kernel_types:
- gemm
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 41545e5bd8be737a3969cd467cd5d88d9e3b8dd0
inclusion_reason: kernel file changes
changed_paths:
- csrc/tinygemm2.cu
- flashinfer/gemm/__init__.py
- flashinfer/gemm/routergemm.py
- flashinfer/jit/__init__.py
- flashinfer/jit/tinygemm2.py
- tests/model_optimizations/test_tinygemm2.py
head_ref_oid: 55595b7e830fad242e06bb1d3b90b840ca64f4de
artifact_dir: evidence/pull-bundles/flashinfer/gh-2587
---
# feat: trtllm tinygemm2 in flashinfer as bf16 routergemm

- Repository: `flashinfer-ai/flashinfer`
- PR: [2587](https://github.com/flashinfer-ai/flashinfer/pull/2587)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `csrc/tinygemm2.cu`
- `flashinfer/gemm/__init__.py`
- `flashinfer/gemm/routergemm.py`
- `flashinfer/jit/__init__.py`
- `flashinfer/jit/tinygemm2.py`
- `tests/model_optimizations/test_tinygemm2.py`

## Evidence Bundle

`evidence/pull-bundles/flashinfer/gh-2587`
