[CI] Fix fused EH norm CI registration (#29845)
This commit is contained in:
@@ -6,7 +6,9 @@ from sglang.jit_kernel.benchmark import marker
|
|||||||
from sglang.jit_kernel.fused_eh_norm import fused_eh_norm
|
from sglang.jit_kernel.fused_eh_norm import fused_eh_norm
|
||||||
from sglang.test.ci.ci_register import register_cuda_ci
|
from sglang.test.ci.ci_register import register_cuda_ci
|
||||||
|
|
||||||
register_cuda_ci(est_time=6, suite="base-b-kernel-benchmark-1-gpu-large")
|
register_cuda_ci(
|
||||||
|
est_time=6, stage="base-b-kernel-benchmark", runner_config="1-gpu-large"
|
||||||
|
)
|
||||||
|
|
||||||
EPS = 1e-6
|
EPS = 1e-6
|
||||||
|
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import torch
|
|||||||
from sglang.jit_kernel.fused_eh_norm import fused_eh_norm
|
from sglang.jit_kernel.fused_eh_norm import fused_eh_norm
|
||||||
from sglang.test.ci.ci_register import register_cuda_ci
|
from sglang.test.ci.ci_register import register_cuda_ci
|
||||||
|
|
||||||
register_cuda_ci(est_time=45, suite="base-b-kernel-unit-1-gpu-large")
|
register_cuda_ci(est_time=45, stage="base-b-kernel-unit", runner_config="1-gpu-large")
|
||||||
register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
|
register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
|
||||||
|
|
||||||
pytestmark = pytest.mark.skipif(
|
pytestmark = pytest.mark.skipif(
|
||||||
|
|||||||
Reference in New Issue
Block a user