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.
flashinfer-ai · Jun 30, 2024 · fec77d0 · fec77d0
1 parent 80a376f
commit fec77d0
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 7 deletions.
diff --git 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
@@ -357,10 +357,6 @@ def __init__(self, *args, **kwargs) -> None:
                     "1",
                     "-Xfatbin",
                     "-compress-all",
-                    "-Xcompiler",
-                    "-mcmodel=medium",
-                    "-Xcompiler",
-                    '"-Wl,--no-relax"',
                 ],
             },
         )