Skip Hopper-only kernels in CI (#3184)

2026-02-27 23:22:08 -08:00
parent 1e3736b19d
commit 6482d13dd3
2 changed files with 5 additions and 3 deletions
@@ -20,7 +20,7 @@ runs:
      run: |
        pip install auditwheel build patchelf setuptools
        python setup.py clean --all
-        MLX_BUILD_STAGE=2 python -m build -w
+        MLX_DISABLE_SM90A_KERNELS=1 MLX_BUILD_STAGE=2 python -m build -w

        auditwheel repair dist/mlx_cuda*.whl \
          --plat manylinux_2_35_${{ inputs.arch }} \