mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
74f3dcab62
Co-authored-by: Costin Eseanu <costineseanu@gmail.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
16 lines
261 B
Batchfile
16 lines
261 B
Batchfile
@echo off
|
|
|
|
set CUDA_HOME=%CUDA_PATH%
|
|
set DISTUTILS_USE_SDK=1
|
|
|
|
set DS_BUILD_AIO=0
|
|
set DS_BUILD_CUTLASS_OPS=0
|
|
set DS_BUILD_EVOFORMER_ATTN=0
|
|
set DS_BUILD_FP_QUANTIZER=0
|
|
set DS_BUILD_RAGGED_DEVICE_OPS=0
|
|
set DS_BUILD_SPARSE_ATTN=0
|
|
|
|
python setup.py bdist_wheel
|
|
|
|
:end
|