From 18c0b5005d986c8e2cad9b0065bc854ec5a43b25 Mon Sep 17 00:00:00 2001
From: rtmadduri <170052006+rtmadduri@users.noreply.github.com>
Date: Thu, 12 Feb 2026 19:02:53 +0000
Subject: [PATCH] chore: Update coverage include list for AMD-modified files

---
 pyproject.toml | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 1cc0322fc8..a750e66ca4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -205,24 +205,23 @@ branch = true
 # AMD/HIP modified files - auto-updated by GitHub Action - DO NOT EDIT THIS SECTION MANUALLY
 include = [
     "flashinfer/__init__.py",
-    "flashinfer/activation.py",
     "flashinfer/aot_hip.py",
     "flashinfer/compilation_context_hip.py",
-    "flashinfer/decode.py",
+    "flashinfer/decode_rocm.py",
     "flashinfer/device_utils.py",
     "flashinfer/get_include_paths.py",
     "flashinfer/hip_utils.py",
     "flashinfer/jit/__init__.py",
-    "flashinfer/jit/activation.py",
     "flashinfer/jit/attention/__init__.py",
     "flashinfer/jit/attention/pytorch_hip.py",
     "flashinfer/jit/core.py",
-    "flashinfer/jit/cpp_ext.py",
+    "flashinfer/jit/cpp_ext_hip.py",
     "flashinfer/jit/env.py",
     "flashinfer/jit/utils.py",
     "flashinfer/norm.py",
     "flashinfer/page.py",
-    "flashinfer/prefill.py",
+    "flashinfer/prefill_rocm.py",
+    "flashinfer/quantization.py",
     "flashinfer/rope.py",
     "flashinfer/sampling.py",
     "flashinfer/utils.py",