From 8f6668d5d42ca901956688f352ccfe602f6f0de6 Mon Sep 17 00:00:00 2001 From: huangkaiyao Date: Thu, 29 May 2025 17:23:43 +0800 Subject: [PATCH] Add SME/SVE matrix operation support. --- llvmorg-19.1.7.tar.gz => llvm-project.tar.gz | 4 ++-- ....0-v1.tar.gz => triton-cpu-3.0.0-v2.tar.gz | 4 ++-- triton-cpu.spec | 20 ++++++++++--------- 3 files changed, 15 insertions(+), 13 deletions(-) rename llvmorg-19.1.7.tar.gz => llvm-project.tar.gz (32%) rename triton-cpu-3.0.0-v1.tar.gz => triton-cpu-3.0.0-v2.tar.gz (32%) diff --git a/llvmorg-19.1.7.tar.gz b/llvm-project.tar.gz similarity index 32% rename from llvmorg-19.1.7.tar.gz rename to llvm-project.tar.gz index 41e92b7..04d274d 100644 --- a/llvmorg-19.1.7.tar.gz +++ b/llvm-project.tar.gz @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:59abea1c22e64933fad4de1671a61cdb934098793c7a31b333ff58dc41bff36c -size 217422846 +oid sha256:93d332a3a649737ae765c0a4e812731be93d2d125d5dac0eb4193d5ca321d7ad +size 224484361 diff --git a/triton-cpu-3.0.0-v1.tar.gz b/triton-cpu-3.0.0-v2.tar.gz similarity index 32% rename from triton-cpu-3.0.0-v1.tar.gz rename to triton-cpu-3.0.0-v2.tar.gz index f0088e5..5a6ac3e 100644 --- a/triton-cpu-3.0.0-v1.tar.gz +++ b/triton-cpu-3.0.0-v2.tar.gz @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:107247893d948660052d966a8051ea2c5b3313c26dc5446be60f8863147114e0 -size 70928347 +oid sha256:9e66238de362abf9cc7cf1720b220dcb7b54f9f548cfdf9cc874e0eb07c2fa86 +size 7002446 diff --git a/triton-cpu.spec b/triton-cpu.spec index 98f49cf..393e247 100644 --- a/triton-cpu.spec +++ b/triton-cpu.spec @@ -1,19 +1,19 @@ Name: triton-cpu Version: 3.0.0 -Release: 2 +Release: 3 Summary: A CPU backend for Triton. License: MIT URL: https://gitee.com/openeuler/triton-cpu -Source0: https://gitee.com/openeuler/triton-cpu/repository/archive/%{name}-%{version}-v1.tar.gz +Source0: https://gitee.com/openeuler/triton-cpu/repository/archive/%{name}-%{version}-v2.tar.gz Source1: triton-downloads.tar.gz -Source2: llvmorg-19.1.7.tar.gz +Source2: llvm-project.tar.gz ExclusiveArch: aarch64 BuildRequires: python3-devel, python3-pip, python3-wheel, python3-lit, python3-pybind11 BuildRequires: python3-setuptools >= 40.8.0 BuildRequires: cmake >= 3.18, cmake < 4.0 BuildRequires: ninja-build >= 1.11.1 -BuildRequires: git, gcc, gcc-c++, zlib-devel +BuildRequires: llvm-toolset-19-clang, zlib-devel %description Triton-CPU is a branch to build a CPU backend for Triton. @@ -21,23 +21,23 @@ Triton-CPU is a branch to build a CPU backend for Triton. %global debug_package %{nil} %prep -tar -zxf %{_sourcedir}/%{name}-%{version}-v1.tar.gz +tar -zxf %{_sourcedir}/%{name}-%{version}-v2.tar.gz unzip -q %{name}/sleef.zip -d %{name}/third_party/ tar -zxf %{_sourcedir}/triton-downloads.tar.gz mkdir %{name}/third_party/nvidia/backend/bin cp -r .triton/nvidia/bin/* %{name}/third_party/nvidia/backend/bin mkdir %{name}/third_party/nvidia/backend/include cp -r .triton/nvidia/include/* %{name}/third_party/nvidia/backend/include -tar -zxf %{_sourcedir}/llvmorg-19.1.7.tar.gz +tar -zxf %{_sourcedir}/llvm-project.tar.gz %build -cd %{_builddir}/llvm-project-llvmorg-19.1.7/ +cd %{_builddir}/llvm-project/ mkdir build && cd build -cmake -G Ninja -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_ASSERTIONS=ON ../llvm -DLLVM_ENABLE_PROJECTS="clang;mlir;llvm;lld" -DLLVM_TARGETS_TO_BUILD="AArch64;NVPTX;AMDGPU" -DLLVM_DEFAULT_TARGET_TRIPLE="aarch64-openEuler-linux" +cmake -G Ninja -DCMAKE_C_COMPILER=/opt/openEuler/llvm-toolset-19/root/usr/bin/clang -DCMAKE_CXX_COMPILER=/opt/openEuler/llvm-toolset-19/root/usr/bin/clang++ -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_ASSERTIONS=ON ../llvm -DLLVM_ENABLE_PROJECTS="clang;compiler-rt;mlir;llvm;lld" -DLLVM_TARGETS_TO_BUILD="AArch64;NVPTX;AMDGPU" -DLLVM_DEFAULT_TARGET_TRIPLE="aarch64-openEuler-linux" ninja %install -export LLVM_BUILD_DIR=%{_builddir}/llvm-project-llvmorg-19.1.7/build +export LLVM_BUILD_DIR=%{_builddir}/llvm-project/build export LLVM_INCLUDE_DIRS=$LLVM_BUILD_DIR/include export LLVM_LIBRARY_DIR=$LLVM_BUILD_DIR/lib export LLVM_SYSPATH=$LLVM_BUILD_DIR @@ -57,6 +57,8 @@ pip install python/ --no-build-isolation --prefix=%{buildroot}%{_prefix} %{python3_sitearch}/triton %changelog +* Thu May 29 2025 huangkaiyao - 3.0.0-3 +- Add SME/SVE matrix operation support * Tue May 20 2025 huangkaiyao - 3.0.0-2 - Move large files to git-lfs * Fri May 9 2025 huangkaiyao - 3.0.0-1 -- Gitee