diff --git a/.gitea/workflows/build-release.yml b/.gitea/workflows/build-release.yml index a6a7d0c..d84804d 100644 --- a/.gitea/workflows/build-release.yml +++ b/.gitea/workflows/build-release.yml @@ -19,6 +19,8 @@ jobs: cuda_home: /usr/local/cuda-13.0 cargo_features: "cuda cudnn flash-attn nccl" compute_caps: "120" + build_jobs: 8 + nvcc_threads: 4 runs-on: ${{ matrix.runner }} steps: - uses: actions/checkout@v4 @@ -53,6 +55,8 @@ jobs: CUDA_HOME: ${{ matrix.cuda_home }} CARGO_FEATURES: ${{ matrix.cargo_features }} CUDA_COMPUTE_CAP: ${{ matrix.compute_caps }} + CARGO_BUILD_JOBS: ${{ matrix.build_jobs }} + NVCC_THREADS: ${{ matrix.nvcc_threads }} SRC_DIR: src - name: Upload binary artifact