Prepare for n_init=auto in KMeans #2896
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: pr | |
| on: | |
| push: | |
| branches: | |
| - "pull-request/[0-9]+" | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| # Please keep pr-builder as the top job here | |
| pr-builder: | |
| needs: | |
| - changed-files | |
| - checks | |
| - clang-tidy | |
| - conda-cpp-build | |
| - conda-cpp-tests | |
| - conda-cpp-checks | |
| - conda-python-build | |
| - conda-python-tests-singlegpu | |
| - conda-python-tests-dask | |
| - conda-notebook-tests | |
| - docs-build | |
| - telemetry-setup | |
| - wheel-build-cuml | |
| - wheel-tests-cuml | |
| - devcontainer | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@branch-24.12 | |
| if: always() | |
| with: | |
| needs: ${{ toJSON(needs) }} | |
| telemetry-setup: | |
| runs-on: ubuntu-latest | |
| continue-on-error: true | |
| env: | |
| OTEL_SERVICE_NAME: "pr-cuml" | |
| steps: | |
| - name: Telemetry setup | |
| uses: rapidsai/shared-actions/telemetry-dispatch-stash-base-env-vars@main | |
| changed-files: | |
| secrets: inherit | |
| needs: telemetry-setup | |
| uses: rapidsai/shared-workflows/.github/workflows/changed-files.yaml@branch-24.12 | |
| with: | |
| files_yaml: | | |
| test_cpp: | |
| - '**' | |
| - '!.devcontainer/**' | |
| - '!.pre-commit-config.yaml' | |
| - '!CONTRIBUTING.md' | |
| - '!README.md' | |
| - '!docs/**' | |
| - '!img/**' | |
| - '!notebooks/**' | |
| - '!python/**' | |
| - '!thirdparty/LICENSES/**' | |
| test_notebooks: | |
| - '**' | |
| - '!.devcontainer/**' | |
| - '!.pre-commit-config.yaml' | |
| - '!CONTRIBUTING.md' | |
| - '!README.md' | |
| - '!thirdparty/LICENSES/**' | |
| test_python: | |
| - '**' | |
| - '!.devcontainer/**' | |
| - '!.pre-commit-config.yaml' | |
| - '!CONTRIBUTING.md' | |
| - '!README.md' | |
| - '!docs/**' | |
| - '!img/**' | |
| - '!notebooks/**' | |
| - '!thirdparty/LICENSES/**' | |
| checks: | |
| secrets: inherit | |
| needs: telemetry-setup | |
| uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@branch-24.12 | |
| with: | |
| enable_check_generated_files: false | |
| ignored_pr_jobs: >- | |
| optional-job-conda-python-tests-cudf-pandas-integration telemetry-summarize | |
| clang-tidy: | |
| needs: checks | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-24.12 | |
| with: | |
| build_type: pull-request | |
| node_type: "cpu8" | |
| arch: "amd64" | |
| container_image: "rapidsai/ci-conda:cuda11.8.0-ubuntu22.04-py3.10" | |
| run_script: "ci/run_clang_tidy.sh" | |
| conda-cpp-build: | |
| needs: checks | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-build.yaml@branch-24.12 | |
| with: | |
| build_type: pull-request | |
| conda-cpp-tests: | |
| needs: [conda-cpp-build, changed-files] | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-tests.yaml@branch-24.12 | |
| if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_cpp | |
| with: | |
| build_type: pull-request | |
| conda-cpp-checks: | |
| needs: conda-cpp-build | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-cpp-post-build-checks.yaml@branch-24.12 | |
| with: | |
| build_type: pull-request | |
| enable_check_symbols: true | |
| symbol_exclusions: raft_cutlass | |
| conda-python-build: | |
| needs: conda-cpp-build | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-python-build.yaml@branch-24.12 | |
| with: | |
| build_type: pull-request | |
| conda-python-tests-singlegpu: | |
| needs: [conda-python-build, changed-files] | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@branch-24.12 | |
| if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python | |
| with: | |
| build_type: pull-request | |
| script: "ci/test_python_singlegpu.sh" | |
| optional-job-conda-python-tests-cudf-pandas-integration: | |
| needs: [conda-python-build, changed-files] | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@branch-24.12 | |
| if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python | |
| with: | |
| matrix_filter: map(select(.ARCH == "amd64")) | |
| build_type: pull-request | |
| script: "ci/test_python_integration.sh" | |
| conda-python-tests-dask: | |
| needs: [conda-python-build, changed-files] | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/conda-python-tests.yaml@branch-24.12 | |
| if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python | |
| with: | |
| build_type: pull-request | |
| script: "ci/test_python_dask.sh" | |
| conda-notebook-tests: | |
| needs: [conda-python-build, changed-files] | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-24.12 | |
| if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_notebooks | |
| with: | |
| build_type: pull-request | |
| node_type: "gpu-v100-latest-1" | |
| arch: "amd64" | |
| container_image: "rapidsai/ci-conda:latest" | |
| run_script: "ci/test_notebooks.sh" | |
| docs-build: | |
| needs: conda-python-build | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-24.12 | |
| with: | |
| build_type: pull-request | |
| node_type: "gpu-v100-latest-1" | |
| arch: "amd64" | |
| container_image: "rapidsai/ci-conda:latest" | |
| run_script: "ci/build_docs.sh" | |
| wheel-build-cuml: | |
| needs: checks | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-24.12 | |
| with: | |
| build_type: pull-request | |
| script: ci/build_wheel.sh | |
| extra-repo: rapidsai/cumlprims_mg | |
| extra-repo-sha: branch-24.12 | |
| extra-repo-deploy-key: CUMLPRIMS_SSH_PRIVATE_DEPLOY_KEY | |
| wheel-tests-cuml: | |
| needs: [wheel-build-cuml, changed-files] | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-24.12 | |
| if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python | |
| with: | |
| build_type: pull-request | |
| script: ci/test_wheel.sh | |
| devcontainer: | |
| needs: telemetry-setup | |
| secrets: inherit | |
| uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@branch-24.12 | |
| with: | |
| arch: '["amd64"]' | |
| cuda: '["12.5"]' | |
| extra-repo-deploy-key: CUMLPRIMS_SSH_PRIVATE_DEPLOY_KEY | |
| build_command: | | |
| sccache -z; | |
| build-all --verbose; | |
| sccache -s; | |
| telemetry-summarize: | |
| runs-on: ubuntu-latest | |
| needs: pr-builder | |
| if: always() | |
| continue-on-error: true | |
| steps: | |
| - name: Load stashed telemetry env vars | |
| uses: rapidsai/shared-actions/telemetry-dispatch-load-base-env-vars@main | |
| with: | |
| load_service_name: true | |
| - name: Telemetry summarize | |
| uses: rapidsai/shared-actions/telemetry-dispatch-write-summary@main | |
| with: | |
| cert_concat: "${{ secrets.OTEL_EXPORTER_OTLP_CA_CERTIFICATE }};${{ secrets.OTEL_EXPORTER_OTLP_CLIENT_CERTIFICATE }};${{ secrets.OTEL_EXPORTER_OTLP_CLIENT_KEY }}" |