Skip to content

[CI][ROCm] Fix aiter sampler fallback test tensor creation on cpu - #62198

Closed
cjackal wants to merge 1 commit into
vllm-project:mainfrom
cjackal:rocm-sampler-wrong-device
Closed

cjackal wants to merge 1 commit into
vllm-project:mainfrom
cjackal:rocm-sampler-wrong-device

Conversation

@cjackal

@cjackal cjackal commented Oct 17, 2026 •

Copy link
Copy Markdown
Contributor

Overview

Fix the CI failures observed on upstream main

Claims

  • Current test_rocm_aiter_sampler_defers_import_when_generators_force_native test is history-dependent; it builds the logits on "default" device, which may not be GPU depending on which shards it is run on. This PR fixes the history dependency by declaring the device explicitly.

Validation

buildkite/ci/pr/amd-mi355-dpx-v1-sample-shard-1-slash-2 CI pipeline passes.

Details

N/A


Pull Request Checklist
  • I used vLLM's /pr-checklist skill. (Mandatory for agents, optional for humans).

  • AI assistance was used during the creation of this PR.

  • Design Fit: Minimizes impact on core components, reuses existing functionality, and justifies added complexity.

  • Testing and Validation: Validates the change and ensures any added tests are meaningful and reliable, with CI coverage or documented CI resource constraints and validation performed outside CI.

  • Code Quality and Style: Keeps code and comments clear and concise, and updates relevant documentation and examples.

  • Pull Request Contents: Includes a brief summary and relevant links, supports claims with evidence, explains root causes and implementation trade-offs, and follows the contributing guide.

Signed-off-by: cjackal <44624812+cjackal@users.noreply.github.com>

@claude claude 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.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@mergify mergify Bot added the rocm Related to AMD ROCm label Oct 17, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD Oct 17, 2026
@cjackal

cjackal commented Oct 18, 2026

Copy link
Copy Markdown
Contributor Author

Closed as fixed by #62213

@cjackal cjackal closed this Oct 18, 2026
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD Oct 18, 2026
@cjackal
cjackal deleted the rocm-sampler-wrong-device branch October 18, 2026 03:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

rocm Related to AMD ROCm

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

1 participant