xmake.lua 12.5 KB
Newer Older
PanZezhongQY's avatar
PanZezhongQY committed
1
add_rules("mode.debug", "mode.release")
2
add_requires("boost", {configs = {stacktrace = true}})
3
4
add_requires("pybind11")

PanZezhongQY's avatar
PanZezhongQY committed
5
6
7
8
9
-- Define color codes
local GREEN = '\27[0;32m'
local YELLOW = '\27[1;33m'
local NC = '\27[0m'  -- No Color

10
set_encodings("utf-8")
PanZezhongQY's avatar
PanZezhongQY committed
11

12
add_includedirs("include")
13
add_includedirs("third_party/spdlog/include")
14

PanZezhongQY's avatar
PanZezhongQY committed
15
16
17
18
if is_mode("debug") then
    add_defines("DEBUG_MODE")
end

19
20
if is_plat("windows") then
    set_runtimes("MD")
21
22
    add_ldflags("/utf-8", {force = true})
    add_cxflags("/utf-8", {force = true})
23
24
end

PanZezhongQY's avatar
PanZezhongQY committed
25
26
27
28
-- CPU
option("cpu")
    set_default(true)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
29
    set_description("Whether to compile implementations for CPU")
PanZezhongQY's avatar
PanZezhongQY committed
30
31
32
option_end()

option("omp")
33
    set_default(true)
PanZezhongQY's avatar
PanZezhongQY committed
34
35
36
37
38
39
40
41
42
    set_showmenu(true)
    set_description("Enable or disable OpenMP support for cpu kernel")
option_end()

if has_config("cpu") then
    includes("xmake/cpu.lua")
    add_defines("ENABLE_CPU_API")
end

43
44
45
46
if has_config("omp") then
    add_defines("ENABLE_OMP")
end

PanZezhongQY's avatar
PanZezhongQY committed
47
48
49
50
-- 英伟达
option("nv-gpu")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
51
    set_description("Whether to compile implementations for Nvidia GPU")
PanZezhongQY's avatar
PanZezhongQY committed
52
53
54
option_end()

if has_config("nv-gpu") then
55
    add_defines("ENABLE_NVIDIA_API")
56
    includes("xmake/nvidia.lua")
PanZezhongQY's avatar
PanZezhongQY committed
57
58
end

59
60
option("cudnn")
    set_default(true)
YdrMaster's avatar
YdrMaster committed
61
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
62
    set_description("Whether to compile cudnn for Nvidia GPU")
YdrMaster's avatar
YdrMaster committed
63
64
option_end()

65
66
if has_config("cudnn") then
    add_defines("ENABLE_CUDNN_API")
YdrMaster's avatar
YdrMaster committed
67
68
end

69
70
71
72
73
74
75
76
77
78
option("cutlass")
    set_default(false)
    set_showmenu(true)
    set_description("Whether to compile cutlass for Nvidia GPU")
option_end()

if has_config("cutlass") then 
    add_defines("ENABLE_CUTLASS_API")
end

79
80
81
82
83
84
85
option("cuda_arch")
    set_showmenu(true)
    set_description("Set CUDA GPU architecture (e.g. sm_90)")
    set_values("sm_50", "sm_60", "sm_70", "sm_75", "sm_80", "sm_86", "sm_89", "sm_90", "sm_90a")
    set_category("option")
option_end()

PanZezhongQY's avatar
PanZezhongQY committed
86
87
88
89
-- 寒武纪
option("cambricon-mlu")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
90
    set_description("Whether to compile implementations for Cambricon MLU")
PanZezhongQY's avatar
PanZezhongQY committed
91
92
93
94
option_end()

if has_config("cambricon-mlu") then
    add_defines("ENABLE_CAMBRICON_API")
PanZezhong's avatar
PanZezhong committed
95
    includes("xmake/bang.lua")
PanZezhongQY's avatar
PanZezhongQY committed
96
97
98
99
100
101
end

-- 华为昇腾
option("ascend-npu")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
102
    set_description("Whether to compile implementations for Huawei Ascend NPU")
PanZezhongQY's avatar
PanZezhongQY committed
103
104
105
106
option_end()

if has_config("ascend-npu") then
    add_defines("ENABLE_ASCEND_API")
Pan Zezhong's avatar
Pan Zezhong committed
107
    includes("xmake/ascend.lua")
PanZezhongQY's avatar
PanZezhongQY committed
108
109
end

110
111
112
113
-- 天数智芯
option("iluvatar-gpu")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
114
    set_description("Whether to compile implementations for Iluvatar GPU")
115
116
117
option_end()

if has_config("iluvatar-gpu") then
118
    add_defines("ENABLE_ILUVATAR_API")
119
120
121
    includes("xmake/iluvatar.lua")
end

122
123
124
125
126
127
128
129
130
131
132
133
-- qy
option("qy-gpu")
    set_default(false)
    set_showmenu(true)
    set_description("Whether to compile implementations for Qy GPU")
option_end()

if has_config("qy-gpu") then
    add_defines("ENABLE_QY_API")
    includes("xmake/qy.lua")
end

PanZezhongQY's avatar
PanZezhongQY committed
134
135
136
137
-- 沐曦
option("metax-gpu")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
138
    set_description("Whether to compile implementations for MetaX GPU")
PanZezhongQY's avatar
PanZezhongQY committed
139
140
option_end()

141
142
143
144
145
146
option("use-mc")
    set_default(false)
    set_showmenu(true)
    set_description("Use MC version")
option_end()

PanZezhongQY's avatar
PanZezhongQY committed
147
if has_config("metax-gpu") then
148
    add_defines("ENABLE_METAX_API")
149
150
151
    if has_config("use-mc") then
        add_defines("ENABLE_METAX_MC_API")
    end
152
    includes("xmake/metax.lua")
PanZezhongQY's avatar
PanZezhongQY committed
153
154
155
156
157
158
end

-- 摩尔线程
option("moore-gpu")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
159
    set_description("Whether to compile implementations for Moore Threads GPU")
PanZezhongQY's avatar
PanZezhongQY committed
160
161
option_end()

162
if has_config("moore-gpu") then
163
    add_defines("ENABLE_MOORE_API")
164
    includes("xmake/moore.lua")
165
end
PanZezhongQY's avatar
PanZezhongQY committed
166

167
168
-- 海光DCU
option("hygon-dcu")
PanZezhongQY's avatar
PanZezhongQY committed
169
170
    set_default(false)
    set_showmenu(true)
171
    set_description("Whether to compile implementations for Hygon DCU")
PanZezhongQY's avatar
PanZezhongQY committed
172
173
option_end()

174
175
176
if has_config("hygon-dcu") then
    add_defines("ENABLE_HYGON_API")
    includes("xmake/hygon.lua")
PanZezhongQY's avatar
PanZezhongQY committed
177
178
end

179
180
181
182
183
184
185
186
187
188
189
-- 昆仑芯
option("kunlun-xpu")
    set_default(false)
    set_showmenu(true)
    set_description("Enable or disable Kunlun XPU kernel")
option_end()

if has_config("kunlun-xpu") then
    add_defines("ENABLE_KUNLUN_API")
    includes("xmake/kunlun.lua")
end
PanZezhongQY's avatar
PanZezhongQY committed
190

191
192
193
194
195
196
197
198
199
200
201
-- 九齿
option("ninetoothed")
    set_default(false)
    set_showmenu(true)
    set_description("Whether to complie NineToothed implementations")
option_end()

if has_config("ninetoothed") then
    add_defines("ENABLE_NINETOOTHED")
end

202
203
204
205
-- InfiniCCL
option("ccl")
    set_default(false)
    set_showmenu(true)
YdrMaster's avatar
YdrMaster committed
206
    set_description("Wether to compile implementations for InfiniCCL")
207
208
209
210
211
212
option_end()

if has_config("ccl") then
    add_defines("ENABLE_CCL")
end

PanZezhong's avatar
PanZezhong committed
213
214
target("infini-utils")
    set_kind("static")
PanZezhong's avatar
PanZezhong committed
215
    on_install(function (target) end)
PanZezhong's avatar
PanZezhong committed
216
    set_languages("cxx17")
217
218
219
220
221
222
223
224
225
226
227
228

    set_warnings("all", "error")

    if is_plat("windows") then
        add_cxflags("/wd4068")
        if has_config("omp") then
            add_cxflags("/openmp")
        end
    else
        add_cxflags("-fPIC", "-Wno-unknown-pragmas")
        if has_config("omp") then
            add_cxflags("-fopenmp")
PanZezhong's avatar
PanZezhong committed
229
            add_ldflags("-fopenmp", {force = true})
230
231
232
        end
    end

PanZezhong's avatar
PanZezhong committed
233
234
235
    add_files("src/utils/*.cc")
target_end()

236
237
238
239
240
241
242
target("infinirt")
    set_kind("shared")

    if has_config("cpu") then
        add_deps("infinirt-cpu")
    end
    if has_config("nv-gpu") then
243
        add_deps("infinirt-nvidia")
244
    end
245
246
247
    if has_config("cambricon-mlu") then
        add_deps("infinirt-cambricon")
    end
248
249
250
    if has_config("ascend-npu") then
        add_deps("infinirt-ascend")
    end
251
252
253
    if has_config("metax-gpu") then
        add_deps("infinirt-metax")
    end
qinyiqun's avatar
qinyiqun committed
254
255
256
    if has_config("moore-gpu") then
        add_deps("infinirt-moore")
    end
YdrMaster's avatar
YdrMaster committed
257
258
259
    if has_config("iluvatar-gpu") then
        add_deps("infinirt-iluvatar")
    end
260
261
262
263
    if has_config("qy-gpu") then
        add_deps("infinirt-qy")
        add_files("build/.objs/infinirt-qy/rules/qy.cuda/src/infinirt/cuda/*.cu.o", {public = true})
    end
zhangyue's avatar
zhangyue committed
264
265
266
    if has_config("kunlun-xpu") then
        add_deps("infinirt-kunlun")
    end
267
268
269
    if has_config("hygon-dcu") then
        add_deps("infinirt-hygon")
    end
270
271
272
    set_languages("cxx17")
    set_installdir(os.getenv("INFINI_ROOT") or (os.getenv(is_host("windows") and "HOMEPATH" or "HOME") .. "/.infini"))
    add_files("src/infinirt/*.cc")
273
    add_installfiles("include/infinirt.h", {prefixdir = "include"})
274
275
target_end()

PanZezhongQY's avatar
PanZezhongQY committed
276
277
target("infiniop")
    set_kind("shared")
278
279
    add_deps("infinirt")

PanZezhongQY's avatar
PanZezhongQY committed
280
281
282
283
    if has_config("cpu") then
        add_deps("infiniop-cpu")
    end
    if has_config("nv-gpu") then
284
        add_deps("infiniop-nvidia")
PanZezhongQY's avatar
PanZezhongQY committed
285
    end
YdrMaster's avatar
YdrMaster committed
286
287
288
    if has_config("iluvatar-gpu") then
        add_deps("infiniop-iluvatar")
    end
289
290
291
292
293
    if has_config("qy-gpu") then
        add_deps("infiniop-qy")
        add_files("build/.objs/infiniop-qy/rules/qy.cuda/src/infiniop/ops/*/nvidia/*.cu.o", {public = true})
        add_files("build/.objs/infiniop-qy/rules/qy.cuda/src/infiniop/devices/nvidia/*.cu.o", {public = true})
    end
PanZezhongQY's avatar
PanZezhongQY committed
294
295

    if has_config("cambricon-mlu") then
PanZezhong's avatar
PanZezhong committed
296
        add_deps("infiniop-cambricon")
PanZezhongQY's avatar
PanZezhongQY committed
297
298
    end
    if has_config("ascend-npu") then
Pan Zezhong's avatar
Pan Zezhong committed
299
        add_deps("infiniop-ascend")
PanZezhongQY's avatar
PanZezhongQY committed
300
301
    end
    if has_config("metax-gpu") then
302
        add_deps("infiniop-metax")
PanZezhongQY's avatar
PanZezhongQY committed
303
    end
304
    if has_config("moore-gpu") then
305
        add_deps("infiniop-moore")
306
    end
307
308
309
    if has_config("kunlun-xpu") then
        add_deps("infiniop-kunlun")
    end
310
311
312
    if has_config("hygon-dcu") then
        add_deps("infiniop-hygon")
    end
PanZezhongQY's avatar
PanZezhongQY committed
313
314
315
316
317
318
319
320
321
322
323
    set_languages("cxx17")
    add_files("src/infiniop/devices/handle.cc")
    add_files("src/infiniop/ops/*/operator.cc")
    add_files("src/infiniop/*.cc")

    set_installdir(os.getenv("INFINI_ROOT") or (os.getenv(is_host("windows") and "HOMEPATH" or "HOME") .. "/.infini"))
    add_installfiles("include/infiniop/(**/*.h)", {prefixdir = "include/infiniop"})
    add_installfiles("include/infiniop/*.h", {prefixdir = "include/infiniop"})
    add_installfiles("include/infiniop.h", {prefixdir = "include"})
    add_installfiles("include/infinicore.h", {prefixdir = "include"})
target_end()
324

325
326
327
328
329
target("infiniccl")
    set_kind("shared")
    add_deps("infinirt")

    if has_config("nv-gpu") then
330
        add_deps("infiniccl-nvidia")
331
    end
Pan Zezhong's avatar
Pan Zezhong committed
332
333
334
    if has_config("ascend-npu") then
        add_deps("infiniccl-ascend")
    end
wooway777's avatar
wooway777 committed
335
336
337
    if has_config("cambricon-mlu") then
        add_deps("infiniccl-cambricon")
    end
PanZezhong's avatar
PanZezhong committed
338
339
340
    if has_config("metax-gpu") then
        add_deps("infiniccl-metax")
    end
YdrMaster's avatar
YdrMaster committed
341
342
343
    if has_config("iluvatar-gpu") then
        add_deps("infiniccl-iluvatar")
    end
344
345
    if has_config("qy-gpu") then
        add_deps("infiniccl-qy")
346
        add_files("build/.objs/infiniccl-qy/rules/qy.cuda/src/infiniccl/cuda/*.cu.o", {public = true})
347
    end
YdrMaster's avatar
YdrMaster committed
348

spike-zhu's avatar
spike-zhu committed
349
350
351
    if has_config("moore-gpu") then
        add_deps("infiniccl-moore")
    end
352

353
354
355
    if has_config("kunlun-xpu") then
        add_deps("infiniccl-kunlun")
    end
356
357
358
    if has_config("hygon-dcu") then
        add_deps("infiniccl-hygon")
    end
359

360
361
362
363
364
365
366
    set_languages("cxx17")

    add_files("src/infiniccl/*.cc")
    add_installfiles("include/infiniccl.h", {prefixdir = "include"})

    set_installdir(os.getenv("INFINI_ROOT") or (os.getenv(is_host("windows") and "HOMEPATH" or "HOME") .. "/.infini"))
target_end()
367

368
target("infinicore_c_api")
369
    set_kind("phony")
370
    add_deps("infiniop", "infinirt", "infiniccl")
371
372
    after_build(function (target) print(YELLOW .. "[Congratulations!] Now you can install the libraries with \"xmake install\"" .. NC) end)
target_end()
373

374
375
376
377
target("infinicore_cpp_api")
    set_kind("shared")
    add_deps("infiniop", "infinirt", "infiniccl")
    set_languages("cxx17")
378
    set_symbols("visibility")
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394

    local INFINI_ROOT = os.getenv("INFINI_ROOT") or (os.getenv(is_host("windows") and "HOMEPATH" or "HOME") .. "/.infini")

    add_includedirs("include")
    add_includedirs(INFINI_ROOT.."/include", { public = true })

    add_linkdirs(INFINI_ROOT.."/lib")
    add_links("infiniop", "infinirt", "infiniccl")

    -- Add InfiniCore C++ source files (needed for RoPE and other nn modules)
    add_files("src/infinicore/*.cc")
    add_files("src/infinicore/context/*.cc")
    add_files("src/infinicore/context/*/*.cc")
    add_files("src/infinicore/tensor/*.cc")
    add_files("src/infinicore/nn/*.cc")
    add_files("src/infinicore/ops/*/*.cc")
395
    add_files("src/utils/*.cc")
396
397
398
399
400
401
402
403
404
405
406

    set_installdir(INFINI_ROOT)
    add_installfiles("include/infinicore/(**.h)",    {prefixdir = "include/infinicore"})
    add_installfiles("include/infinicore/(**.hpp)",    {prefixdir = "include/infinicore"})
    add_installfiles("include/infinicore/(**/*.h)",  {prefixdir = "include/infinicore"})
    add_installfiles("include/infinicore/(**/*.hpp)",{prefixdir = "include/infinicore"})
    add_installfiles("include/infinicore.h",          {prefixdir = "include"})
    add_installfiles("include/infinicore.hpp",        {prefixdir = "include"})
    after_build(function (target) print(YELLOW .. "[Congratulations!] Now you can install the libraries with \"xmake install\"" .. NC) end)
target_end()

407
408
409
410
411
412
413
414
415
416
target("_infinicore")
    add_packages("boost")
    if is_mode("debug") then
        add_defines("BOOST_STACKTRACE_USE_BACKTRACE")
        add_links("backtrace")
    else
        add_defines("BOOST_STACKTRACE_USE_NOOP")
    end

    set_default(false)
417
418
    add_rules("python.library", {soabi = true})
    add_packages("pybind11")
419
    set_languages("cxx17")
420
421

    set_kind("shared")
422
423
424
425
426
    local INFINI_ROOT = os.getenv("INFINI_ROOT") or (os.getenv(is_host("windows") and "HOMEPATH" or "HOME") .. "/.infini")
    add_includedirs(INFINI_ROOT.."/include", { public = true })

    add_linkdirs(INFINI_ROOT.."/lib")
    add_links("infiniop", "infinirt", "infiniccl")
427
428

    add_files("src/infinicore/*.cc")
429
430
431
    add_files("src/infinicore/context/*.cc")
    add_files("src/infinicore/context/*/*.cc")
    add_files("src/infinicore/tensor/*.cc")
432
    add_files("src/infinicore/nn/*.cc")
433
    add_files("src/infinicore/ops/*/*.cc")
434
    add_files("src/infinicore/pybind11/**.cc")
435
    add_files("src/utils/*.cc")
436

437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
    set_installdir("python/infinicore")
target_end()

option("editable")
    set_default(false)
    set_showmenu(true)
    set_description("Install the `infinicore` Python package in editable mode")
option_end()

target("infinicore")
    set_kind("phony")

    set_default(false)

    add_deps("_infinicore")

    on_install(function (target)
        local pip_install_args = {}

        if has_config("editable") then
            table.insert(pip_install_args, "--editable")
        end

        os.execv("python", table.join({"-m", "pip", "install"}, pip_install_args, {"."}))
    end)
462
463
target_end()

464
465
-- Tests
includes("xmake/test.lua")