Skip to content

ci: probe host GPU driver version in sandbox test logs - #8819

Closed
delock wants to merge 2 commits into
masterfrom
ci/driver-probe
Closed

delock wants to merge 2 commits into
masterfrom
ci/driver-probe

Conversation

@delock

@delock delock commented Oct 14, 2026

Copy link
Copy Markdown
Collaborator

Diagnostic probe for #8810 / merge-queue hang wave: prints host driver version (nvidia-smi) inside the Modal sandbox so infra regressions (suspected fleet driver update to 610.57.04) are visible in CI logs. Also adds a small CUDA context health test.

Signed-off-by: Ma, Guokai <guokai.ma@intel.com>
Signed-off-by: Ma, Guokai <guokai.ma@intel.com>

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 52d7f4d793

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".



def _driver_version() -> str:
result = subprocess.run(["nvidia-smi", "--query-gpu=driver_version,name", "--format=csv,noheader"],

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

P1 Badge Skip the NVIDIA probe on non-NVIDIA jobs

When this file is collected by the required CPU workflow (.github/workflows/cpu-torch-latest.yml:252-257 runs all of unit/), nvidia-smi is not installed, but this call executes unconditionally and raises FileNotFoundError; the CUDA availability guard in the second test never protects this test. Gate the probe on an NVIDIA environment or move it into the NVIDIA-only sandbox controller.

Useful? React with 👍 / 👎.

Comment on lines +24 to +25
def test_report_host_driver_version():
print(f"DS_CI_DRIVER_PROBE: {_driver_version()}")

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

P2 Badge Emit the driver report outside pytest capture

In the Modal path, ci/torch_latest.py:739-748 invokes pytest with --verbose but not -s; pytest --help documents -s as the shortcut for --capture=no and says captured output is shown for failed tests. Because this test normally passes, the printed driver version is suppressed and therefore does not become visible in the CI log as intended. Emit it as an unconditional controller command before pytest or explicitly disable capture for the probe.

Useful? React with 👍 / 👎.

Comment on lines +24 to +25
def test_report_host_driver_version():
print(f"DS_CI_DRIVER_PROBE: {_driver_version()}")

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

P2 Badge Run the probe for every sandbox invocation

After this commit is merged, normal merge-queue runs use subset selection and execute exactly the impacted test files (.github/workflows/TEST_SELECTION.md:176-188). The selector only includes this standalone test when the test itself changes (ci/tests_fetcher.py:543-567), so unrelated future changes—the runs for which a fleet driver regression must be diagnosed—will not execute the probe at all. Put the report in the sandbox controller's unconditional pre-pytest commands rather than relying on test selection.

Useful? React with 👍 / 👎.

@chatgpt-codex-connector

Copy link
Copy Markdown

Codex Review Summary

This comment shows the latest Codex review activity on this pull request.

Review Status Commit Review trigger
📝 Code Review ✅ Completed 2026-10-14T06:57:57.283116Z 52d7f4d PR opened
ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review" or "@codex security review".

Codex reacts with 👀 while any review is running, comments if it has suggestions, and reacts with 👍 once all reviews finish with no findings.

@delock
delock enabled auto-merge October 14, 2026 06:59
@delock
delock disabled auto-merge October 14, 2026 07:10
@delock delock closed this Oct 14, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant