|
63 | 63 |
|
64 | 64 | - label: Async Engine, Inputs, Utils, Worker Test (CPU) # 4 mins |
65 | 65 | timeout_in_minutes: 10 |
66 | | - mirror_hardwares: [amdexperimental] |
| 66 | + mirror_hardwares: [amdexperimental, amdproduction] |
67 | 67 | agent_pool: mi325_1 |
68 | 68 | # grade: Blocking |
69 | 69 | source_file_dependencies: |
@@ -353,7 +353,7 @@ steps: |
353 | 353 | - pytest -v -s entrypoints/openai/correctness/test_lmeval.py::test_lm_eval_accuracy_v1_engine |
354 | 354 |
|
355 | 355 | - label: V1 Test others (CPU) # 5 mins |
356 | | - mirror_hardwares: [amdexperimental] |
| 356 | + mirror_hardwares: [amdexperimental, amdproduction] |
357 | 357 | agent_pool: mi325_1 |
358 | 358 | # grade: Blocking |
359 | 359 | source_file_dependencies: |
@@ -459,6 +459,7 @@ steps: |
459 | 459 | - pytest -v -s compile/test_fusion_all_reduce.py |
460 | 460 | - pytest -v -s compile/test_decorator.py |
461 | 461 | - pytest -v -s compile/test_noop_elimination.py |
| 462 | + - pytest -v -s compile/test_aot_compile.py |
462 | 463 |
|
463 | 464 | - label: PyTorch Fullgraph Smoke Test # 15min |
464 | 465 | timeout_in_minutes: 30 |
@@ -487,14 +488,14 @@ steps: |
487 | 488 |
|
488 | 489 | - label: Kernels Core Operation Test # 48min |
489 | 490 | timeout_in_minutes: 75 |
490 | | - mirror_hardwares: [amdexperimental] |
| 491 | + mirror_hardwares: [amdexperimental, amdproduction] |
491 | 492 | agent_pool: mi325_1 |
492 | 493 | # grade: Blocking |
493 | 494 | source_file_dependencies: |
494 | 495 | - csrc/ |
495 | 496 | - tests/kernels/core |
496 | 497 | commands: |
497 | | - - pytest -v -s kernels/core |
| 498 | + - pytest -v -s kernels/core kernels/test_top_k_per_row.py |
498 | 499 |
|
499 | 500 | - label: Kernels Attention Test %N # 23min |
500 | 501 | timeout_in_minutes: 35 |
@@ -632,7 +633,7 @@ steps: |
632 | 633 |
|
633 | 634 | - label: OpenAI-Compatible Tool Use # 23 min |
634 | 635 | timeout_in_minutes: 35 |
635 | | - mirror_hardwares: [amdexperimental] |
| 636 | + mirror_hardwares: [amdexperimental, amdproduction] |
636 | 637 | agent_pool: mi325_1 |
637 | 638 | # grade: Blocking |
638 | 639 | fast_check: false |
|
0 commit comments