diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index d0ee89966..4b5a17e0c 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -13,7 +13,6 @@ # limitations under the License. # ============================================================================== -import multiprocessing import os import sys from pathlib import Path @@ -175,10 +174,6 @@ setup( packages=find_packages(), package_dir={"": "src"}, ext_modules=ext_modules, - cmdclass={ - "build_ext": BuildExtension.with_options( - use_ninja=True, max_jobs=multiprocessing.cpu_count() - ) - }, + cmdclass={"build_ext": BuildExtension.with_options(use_ninja=True)}, options={"bdist_wheel": {"py_limited_api": "cp39"}}, )