[CI] Shard (H100) Helion Kernels five ways (#58645)
Signed-off-by: Thang Nguyen <thangnguyenvn647@gmail.com> Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5.5
parent
afea5c20c7
commit
fa6c407d15
@@ -601,13 +601,15 @@ steps:
|
||||
commands:
|
||||
- pip install helion==1.4.0
|
||||
- pytest -v -s kernels/helion/ --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
|
||||
parallelism: 2
|
||||
parallelism: 5
|
||||
mirror:
|
||||
amd:
|
||||
label: ":amd: (MI250) Helion Kernels Shard %N"
|
||||
dind: false
|
||||
device: mi250_1
|
||||
timeout_in_minutes: 45
|
||||
# Explicit so the base step's count does not multiply the mirror's replicas.
|
||||
parallelism: 2
|
||||
depends_on:
|
||||
- image-build-amd
|
||||
source_file_dependencies:
|
||||
|
||||
Reference in New Issue
Block a user