Skip to content

Commit f7c55cb

Browse files
authored
Fix special_chebyshev_polynomial_w ut (flagos-ai#5352)
1 parent 7192eb8 commit f7c55cb

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/flag_gems/ops/special_chebyshev_polynomial_w.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
logger = logging.getLogger(__name__)
2525

2626
# Maximum supported polynomial degree (arbitrary but safe bound)
27-
_MAX_POLY_DEGREE: tl.constexpr = 101
27+
_MAX_POLY_DEGREE = tl.constexpr(101)
2828

2929

3030
@pointwise_dynamic(promotion_methods=[(0, 1, "INT_TO_FLOAT")])

tests/test_special_chebyshev_polynomial_w.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def test_special_chebyshev_polynomial_w_out_of_domain(dtype):
6161
utils.gems_assert_close(res_out, ref_out, dtype, equal_nan=True)
6262

6363

64-
@pytest.mark.special_chebyshev_polynomial_w
64+
@pytest.mark.special_chebyshev_polynomial_w_out
6565
@pytest.mark.parametrize("shape", utils.SPECIAL_SHAPES)
6666
@pytest.mark.parametrize("dtype", [torch.float32])
6767
def test_special_chebyshev_polynomial_w_out(shape, dtype):

0 commit comments

Comments
 (0)