---
id: pr-flashinfer-1481
repo: flashinfer-ai/flashinfer
pr: 1481
title: Add python API for masked grouped gemm
author: kaixih
date: '2025-08-14'
url: https://github.com/flashinfer-ai/flashinfer/pull/1481
source_category: upstream-code
architectures:
- sm100
tags:
- fp4
- gemm
techniques: []
hardware_features:
- fp4
kernel_types:
- gemm
languages:
- cute-dsl
- python
captured_at: '2026-04-27'
status: merged
merge_sha: a2e5ea665a55e2cf1eae1a26b5aec48153715464
inclusion_reason: kernel file changes
changed_paths:
- flashinfer/cute_dsl/blockscaled_gemm.py
- flashinfer/cute_dsl/utils.py
- flashinfer/utils.py
- pyproject.toml
- tests/test_cute_dsl_blockscaled_gemm.py
- tests/test_fp4_tensor_torch_cute.py
head_ref_oid: 0797d8a071168ffe4c1f8cf3720c33f93d0fe5b2
artifact_dir: evidence/pull-bundles/flashinfer/gh-1481
---
# Add python API for masked grouped gemm

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

## Changed Paths

- `flashinfer/cute_dsl/blockscaled_gemm.py`
- `flashinfer/cute_dsl/utils.py`
- `flashinfer/utils.py`
- `pyproject.toml`
- `tests/test_cute_dsl_blockscaled_gemm.py`
- `tests/test_fp4_tensor_torch_cute.py`

## Evidence Bundle

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