%global commit 0.8.0
%global shortcommit %(c=%{commit}; echo ${c:0:7})

Name:           deepspeed
Version:        0.8.0
Release:        1%{?dist}
Summary:        DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

License:        MIT
URL:            https://www.deepspeed.ai/
Source0:        https://github.com/microsoft/DeepSpeed/archive/v%{version}.tar.gz

BuildRequires:  gcc
BuildRequires:  gcc-c++
BuildRequires:  cmake
BuildRequires:  python3-devel
BuildRequires:  python3-numpy
BuildRequires:  python3-scipy
BuildRequires:  python3-scikit-learn
BuildRequires:  python3-matplotlib
BuildRequires:  python3-setuptools
BuildRequires:  python3-pip
BuildRequires:  python3-wheel
BuildRequires:  python3-pytorch
BuildRequires:  python3-six
BuildRequires:  python3-tqdm
BuildRequires:  python3-requests
BuildRequires:  python3-yaml
BuildRequires:  python3-regex
BuildRequires:  python3-psutil
BuildRequires:  python3-typing-extensions
BuildRequires:  python3-future
BuildRequires:  python3-filelock
BuildRequires:  python3-importlib-metadata
BuildRequires:  python3-packaging
BuildRequires:  python3-dataclasses

%description
DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
It was developed by Microsoft Research and is widely used in the AI community.

%global debug_package %{nil}

%prep
%autosetup -n DeepSpeed-%{commit}

%build
export DEEPSPEED_BUILD_VERSION=%{version}
export DEEPSPEED_BUILD_NUMBER=1
python3 setup.py bdist_wheel

%install
%py3_install

%files
%{python3_sitelib}/*

%changelog
* Sat Apr 15 2024 weilaijishu - 0.8.0-1
- Updated to version 0.8.0