From d27de0657bc228623d3f5c5b1d5a6169e5b922fa Mon Sep 17 00:00:00 2001
From: demandal25 <216094842+demandal25@users.noreply.github.com>
Date: Tue, 16 Jun 2026 05:02:11 +0000
Subject: [PATCH] chore: Update coverage include list for AMD-modified files

---
 pyproject.toml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 7cd3eab95f..11ad0f3d41 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -220,11 +220,14 @@ include = [
     "flashinfer/hip_utils.py",
     "flashinfer/jit/__init__.py",
     "flashinfer/jit/activation.py",
+    "flashinfer/jit/aiter_source.py",
     "flashinfer/jit/attention/__init__.py",
     "flashinfer/jit/attention/modules_hip.py",
     "flashinfer/jit/core.py",
     "flashinfer/jit/cpp_ext_hip.py",
     "flashinfer/jit/env.py",
+    "flashinfer/jit/norm.py",
+    "flashinfer/jit/rope.py",
     "flashinfer/jit/utils.py",
     "flashinfer/logits_processor/__init__.py",
     "flashinfer/mla_rocm.py",