diff --git a/sgl-kernel/pyproject.toml b/sgl-kernel/pyproject.toml index d07fda861..278af1e46 100644 --- a/sgl-kernel/pyproject.toml +++ b/sgl-kernel/pyproject.toml @@ -1,7 +1,7 @@ [build-system] requires = [ "scikit-build-core>=0.10", - "torch==2.5.1", + "torch>=2.5.1", "wheel", ] build-backend = "scikit_build_core.build" diff --git a/sgl-kernel/pyproject_rocm.toml b/sgl-kernel/pyproject_rocm.toml index bab23cfdb..c13f3f026 100644 --- a/sgl-kernel/pyproject_rocm.toml +++ b/sgl-kernel/pyproject_rocm.toml @@ -2,7 +2,7 @@ requires = [ "setuptools>=75.0", "scikit-build-core>=0.10", - "torch==2.5.1", + "torch>=2.5.1", "wheel", ] build-backend = "setuptools.build_meta"