Skip to content

docs: update readme makefile #16

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
bitzyz merged 1 commit into from
Feb 10, 2026
Merged

docs: update readme makefile #16

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand All @@ -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)
Expand Down
22 changes: 10 additions & 12 deletions Makefile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,24 +2,22 @@

TYPE ?= Release
TEST ?= ON
# 平台参数（CUDA / ASCEND / CPU / ...）
PLATFORM ?= CPU
USE_CUDA ?= OFF
USE_ASCEND ?= OFF
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)
Expand All @@ -37,17 +35,17 @@ 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
USE_MOORE = ON
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
Expand All @@ -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
Expand All @@ -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
Expand Down
2 changes: 1 addition & 1 deletion README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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` |

---
Expand Down