%global pypi_name deepspeed
%global pypi_version 0.13.5
%global pkg_summary DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
%global pkg_description %{expand:
%{pkg_summary}}
%global debug_package %{nil}

Name:           python-%{pypi_name}
Version:        %{pypi_version}
Release:        1
Summary:        %{pkg_summary}

License:        Apache-2.0
URL:            https://github.com/microsoft/DeepSpeed
Source0:        %{url}/archive/refs/tags/v%{version}.tar.gz#/DeepSpeed-%{version}.tar.gz

BuildRequires: cmake
BuildRequires: ninja-build
BuildRequires: g++

BuildRequires: python3-devel
BuildRequires: python3-setuptools
BuildRequires: python3-setuptools_scm
BuildRequires: python3-pbr
BuildRequires: python3-pip
BuildRequires: python3-wheel
BuildRequires: python3-hatchling

BuildRequires: python3-pytorch >= 2.1.2

%description
%{pkg_description}

%package -n     python3-%{pypi_name}
Summary:        %{pkg_summary}

%description -n python3-%{pypi_name}
%{pkg_description}

%prep
%autosetup -p1 -n DeepSpeed-%{pypi_version}

%build
%pyproject_build

%install
%pyproject_install

%files -n python3-%{pypi_name}
%license LICENSE
%doc README.md
%{python3_sitelib}/%{pypi_name}*

%changelog
* Sun Mar 31 2024 YunShu
- add DeepSpeed 0.13.5