Skip to content

[CI] Register jit_kernel Test Files to Solve No Registry Found Error#21541

Closed
bingxche wants to merge 1 commit intomainfrom
bingxche/fix-no-registry-found
Closed

[CI] Register jit_kernel Test Files to Solve No Registry Found Error#21541
bingxche wants to merge 1 commit intomainfrom
bingxche/fix-no-registry-found

Conversation

@bingxche
Copy link
Copy Markdown
Collaborator

Motivation

#21538

Modifications

Accuracy Tests

Benchmarking and Profiling

Checklist

Review Process

  1. Ping Merge Oncalls to start the PR flow. See the PR Merge Process.
  2. Get approvals from CODEOWNERS and other reviewers.
  3. Trigger CI tests with comments or contact authorized users to do so.
    • /tag-run-ci-label, /rerun-failed-ci, /tag-and-rerun-ci
  4. After green CI and required approvals, ask Merge Oncalls to merge.

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request integrates JIT kernel benchmarks and unit tests into the CUDA CI pipeline by adding registration calls. The review feedback identifies that the disabled parameter should be removed from these new registrations since estimated execution times are now provided, ensuring the tests actually run during CI.

Comment on lines +14 to +18
register_cuda_ci(
est_time=13,
suite="stage-b-kernel-benchmark-1-gpu-large",
disabled="unknown est_time",
)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The disabled parameter will cause this test to be registered but skipped by the CI. Since an est_time is provided and the goal of this PR is to enable these tests, the disabled parameter should be removed.

register_cuda_ci(
    est_time=13,
    suite="stage-b-kernel-benchmark-1-gpu-large",
)

Comment on lines +23 to +27
register_cuda_ci(
est_time=17,
suite="stage-b-kernel-benchmark-1-gpu-large",
disabled="unknown est_time",
)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The disabled parameter will cause this test to be registered but skipped by the CI. Since an est_time is provided and the goal of this PR is to enable these tests, the disabled parameter should be removed.

register_cuda_ci(
    est_time=17,
    suite="stage-b-kernel-benchmark-1-gpu-large",
)

Comment on lines +7 to +17
register_cuda_ci(
est_time=24,
suite="stage-b-kernel-unit-1-gpu-large",
disabled="unknown est_time",
)
register_cuda_ci(
est_time=120,
suite="nightly-kernel-1-gpu",
nightly=True,
disabled="unknown est_time",
)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The disabled parameter will cause these tests to be registered but skipped by the CI. Since an est_time is provided for both and the goal of this PR is to enable these tests, the disabled parameter should be removed from both register_cuda_ci calls.

register_cuda_ci(
    est_time=24,
    suite="stage-b-kernel-unit-1-gpu-large",
)
register_cuda_ci(
    est_time=120,
    suite="nightly-kernel-1-gpu",
    nightly=True,
)

Comment on lines +14 to +24
register_cuda_ci(
est_time=64,
suite="stage-b-kernel-unit-1-gpu-large",
disabled="unknown est_time",
)
register_cuda_ci(
est_time=256,
suite="nightly-kernel-1-gpu",
nightly=True,
disabled="unknown est_time",
)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The disabled parameter will cause these tests to be registered but skipped by the CI. Since an est_time is provided for both and the goal of this PR is to enable these tests, the disabled parameter should be removed from both register_cuda_ci calls.

register_cuda_ci(
    est_time=64,
    suite="stage-b-kernel-unit-1-gpu-large",
)
register_cuda_ci(
    est_time=256,
    suite="nightly-kernel-1-gpu",
    nightly=True,
)

@Fridge003
Copy link
Copy Markdown
Collaborator

Issue already solved

@Fridge003 Fridge003 closed this Mar 27, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants