Name:           python-colibri-engine
Version:        1.4.0
Release:        %autorelease
Summary:        Run frontier MoE models on hardware you already own

License:        Apache-2.0

URL:            https://github.com/JustVugg/colibri
Source0:        %{url}/archive/v%{version}/v%{version}.tar.gz

ExclusiveArch:  x86_64

BuildRequires:  chrpath
BuildRequires:  gcc
BuildRequires:  gcc-c++
BuildRequires:  hipcc
BuildRequires:  make
BuildRequires:  python3-devel
BuildRequires:  rocm-filesystem
BuildRequires:  rocm-hip-devel
BuildRequires:  rocm-rpm-macros
BuildRequires:  rocm-runtime-devel
BuildRequires:  rocwmma-devel

%description
%{summary}.

%package -n     python3-colibri-engine
Summary:        Run frontier MoE models on hardware you already own

%description -n python3-colibri-engine
%{summary}.

Requires:       nodejs

%prep
%autosetup -p1 -n colibri-%{version}
# Add -g for debuginfo
sed -i -e 's@-O3@-O3 -g@' c/Makefile
# change install location of the real coli
sed -i -e 's@engine_dir = os.path.join(os.path.dirname(here), "c")@engine_dir = "%{_libexecdir}/colibri/c"@' colibri/cli.py

%generate_buildrequires
# orcale needs python3-transformers
%pyproject_buildrequires -x convert,bench

%build
# rocwmma can not run all gpus, here is list ..
GPUS="gfx908,gfx90a,gfx942,gfx950,gfx1100,gfx1101,gfx1102,gfx1151,gfx1150,gfx1200,gfx1201"
%make_build HIP=1 ROCM_HOME=/usr HIP_ARCH=${GPUS}
%pyproject_wheel

%install
%pyproject_install
%pyproject_save_files colibri

mkdir -p %{buildroot}%{_libexecdir}/colibri/c
install c/coli %{buildroot}%{_libexecdir}/colibri/c/
scripts="doctor openai_server resource_plan version"
for s in $scripts; do
    install c/${s}.py %{buildroot}%{_libexecdir}/colibri/c/
done
install c/colibri %{buildroot}%{_libexecdir}/colibri/c/
# ERROR   0001: file '/usr/libexec/colibri/c/colibri' contains a standard runpath '/usr/lib' in [/usr/lib]
chrpath -d %{buildroot}%{_libexecdir}/colibri/c/colibri

%check
%pyproject_check_import

%files -n python3-colibri-engine -f %{pyproject_files}
%doc README.it.md  README.md  README.zh-CN.md  README.zh-TW.md GPU_BACKENDS.md
%license LICENSE
%{_bindir}/coli
%{_libexecdir}/colibri/

%changelog
%autochangelog