From 640c9b399a13beabb7f156ef8f8d2c53d0b3ea3a Mon Sep 17 00:00:00 2001 From: zhangyunze Date: Mon, 9 Feb 2026 09:26:34 +0000 Subject: [PATCH] docs: update readme makefile --- CMakeLists.txt | 6 +++--- Makefile | 22 ++++++++++------------ README.md | 2 +- 3 files changed, 14 insertions(+), 16 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 58eb9ea..acac82b 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -5,7 +5,7 @@ option(USE_CAMBRICON "Use Cambricon" OFF) option(USE_METAX "Use Metax" OFF) option(USE_MOORE "Use Moore" OFF) option(USE_ILUVATAR "Use Iluvatar" OFF) -option(USE_SUGON "Use Sugon" OFF) +option(USE_HYGON "Use Sugon" OFF) option(USE_KUNLUN "Use Kunlun" OFF) cmake_minimum_required(VERSION 3.17) @@ -27,8 +27,8 @@ endif() if(USE_ILUVATAR) add_compile_definitions(USE_ILUVATAR=1) endif() -if(USE_SUGON) - add_compile_definitions(USE_SUGON=1) +if(USE_HYGON) + add_compile_definitions(USE_HYGON=1) endif() if(USE_KUNLUN) add_compile_definitions(USE_KUNLUN=1) diff --git a/Makefile b/Makefile index ec35ab7..1e35b02 100644 --- a/Makefile +++ b/Makefile @@ -2,7 +2,6 @@ TYPE ?= Release TEST ?= ON -# 平台参数(CUDA / ASCEND / CPU / ...) PLATFORM ?= CPU USE_CUDA ?= OFF USE_ASCEND ?= OFF @@ -10,16 +9,15 @@ USE_CAMBRICON ?= OFF USE_METAX ?= OFF USE_MOORE ?= OFF USE_ILUVATAR ?= OFF -USE_SUGON ?= OFF +USE_HYGON ?= OFF USE_KUNLUN ?= OFF -# 通信开关(ON / OFF) COMM ?= OFF FORMAT_ORIGIN ?= CMAKE_OPT = -DCMAKE_BUILD_TYPE=$(TYPE) CMAKE_OPT += -DBUILD_TEST=$(TEST) -# InfiniCore 仓库地址 +# InfiniCore repo address INFINICORE_URL = git@github.com:InfiniTensor/InfiniCore.git INFINICORE_DIR = InfiniCore CUR_DIR := $(shell pwd) @@ -37,7 +35,7 @@ else ifeq ($(PLATFORM), CAMBRICON) XMAKE_PLATFORM_FLAG = --cambricon-mlu=y USE_CAMBRICON = ON else ifeq ($(PLATFORM), METAX) - XMAKE_PLATFORM_FLAG = --metax-gpu=y + XMAKE_PLATFORM_FLAG = --metax-gpu=y --use-mc=y USE_METAX = ON else ifeq ($(PLATFORM), MOORE) XMAKE_PLATFORM_FLAG = --moore-gpu=y @@ -45,9 +43,9 @@ else ifeq ($(PLATFORM), MOORE) else ifeq ($(PLATFORM), ILUVATAR) XMAKE_PLATFORM_FLAG = --iluvatar-gpu=y USE_ILUVATAR = ON -else ifeq ($(PLATFORM), SUGON) - XMAKE_PLATFORM_FLAG = --sugon-dcu=y - USE_SUGON = ON +else ifeq ($(PLATFORM), HYGON) + XMAKE_PLATFORM_FLAG = --hygon-dcu=y + USE_HYGON = ON else ifeq ($(PLATFORM), KUNLUN) XMAKE_PLATFORM_FLAG = --kunlun-xpu=y USE_KUNLUN = ON @@ -61,10 +59,10 @@ CMAKE_OPT += -DUSE_CAMBRICON=$(USE_CAMBRICON) CMAKE_OPT += -DUSE_METAX=$(USE_METAX) CMAKE_OPT += -DUSE_MOORE=$(USE_MOORE) CMAKE_OPT += -DUSE_ILUVATAR=$(USE_ILUVATAR) -CMAKE_OPT += -DUSE_SUGON=$(USE_SUGON) +CMAKE_OPT += -DUSE_HYGON=$(USE_HYGON) CMAKE_OPT += -DUSE_KUNLUN=$(USE_KUNLUN) -# 通信参数 +# communication switch ifeq ($(COMM), ON) XMAKE_COMM_FLAG = --ccl=y else @@ -77,11 +75,11 @@ check-infini: @if [ -z "$$INFINI_ROOT" ]; then \ echo "[INFO] INFINI_ROOT 未设置,开始拉取 InfiniCore ..."; \ if [ ! -d "$(INFINICORE_DIR)" ]; then \ - git clone $(INFINICORE_URL); \ + git clone --recursive $(INFINICORE_URL); \ fi; \ echo "[INFO] 开始安装 InfiniCore (PLATFORM=$(PLATFORM), COMM=$(COMM)) ..."; \ cd $(INFINICORE_DIR) && python scripts/install.py $(XMAKE_FLAGS); \ - echo "[INFO] 请运行 source ./start.sh 设置环境变量"; \ + echo "[INFO] 请手动运行 source ./start.sh 设置环境变量"; \ else \ echo "[INFO] 检测到 INFINI_ROOT=$$INFINI_ROOT"; \ fi diff --git a/README.md b/README.md index 0c696bd..762f5f6 100644 --- a/README.md +++ b/README.md @@ -39,7 +39,7 @@ | METAX | 沐曦 MTX | `USE_METAX=ON` | | MOORE | 摩尔线程 Moore | `USE_MOORE=ON` | | ILUVATAR | 天数智芯 | `USE_ILUVATAR=ON` | -| SUGON | 曙光 DCU | `USE_SUGON=ON` | +| HYGON | 海光 DCU | `USE_HYGON=ON` | | KUNLUN | 百度昆仑 XPU | `USE_KUNLUN=ON` | ---