From fec77d064ebea78c5b811a458ec962f2867103aa Mon Sep 17 00:00:00 2001
From: Zihao Ye <expye@outlook.com>
Date: Sun, 30 Jun 2024 16:31:51 -0700
Subject: [PATCH] ci: remove redundant `NUM_FRAGS_Z` (#345)

Do not compile `NUM_FRAGS_Z=6` to reduce wheel size.
Also revert #341 as they don't make effect.
---
 include/flashinfer/utils.cuh | 3 ---
 python/setup.py              | 4 ----
 2 files changed, 7 deletions(-)

diff --git a/include/flashinfer/utils.cuh b/include/flashinfer/utils.cuh
index 04890373..fa148642 100644
--- a/include/flashinfer/utils.cuh
+++ b/include/flashinfer/utils.cuh
@@ -75,9 +75,6 @@
   if (max_frags_z >= 8) {                                   \
     constexpr size_t NUM_FRAGS_Z = 8;                       \
     __VA_ARGS__                                             \
-  } else if (max_frags_z >= 6) {                            \
-    constexpr size_t NUM_FRAGS_Z = 6;                       \
-    __VA_ARGS__                                             \
   } else if (max_frags_z >= 4) {                            \
     constexpr size_t NUM_FRAGS_Z = 4;                       \
     __VA_ARGS__                                             \
diff --git a/python/setup.py b/python/setup.py
index 45881d7b..f6f8ff48 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -357,10 +357,6 @@ def __init__(self, *args, **kwargs) -> None:
                     "1",
                     "-Xfatbin",
                     "-compress-all",
-                    "-Xcompiler",
-                    "-mcmodel=medium",
-                    "-Xcompiler",
-                    '"-Wl,--no-relax"',
                 ],
             },
         )