From 0544bcce46138a28f9ff7a1a45a38100009fe6f8 Mon Sep 17 00:00:00 2001 From: alter-xp Date: Mon, 9 Dec 2024 10:52:24 +0800 Subject: [PATCH] Support building on the RISC-V platform. When building on the RISC-V platform, remove the dependency on Triton. --- setup.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/setup.py b/setup.py index 14465ba6..b8f8cac0 100644 --- a/setup.py +++ b/setup.py @@ -1,5 +1,6 @@ import os import torch +import platform from pathlib import Path from setuptools import setup, find_packages @@ -45,6 +46,10 @@ "huggingface_hub>=0.26.5", ] +if 'riscv' in platform.machine().lower(): + # remove triton for riscv platform + requirements.remove("triton") + setup( packages=find_packages(), install_requires=requirements,