From 2d936dca1cf30be1bd67aef7ff29f79c3a0a1f2c Mon Sep 17 00:00:00 2001 From: Tim Moon Date: Fri, 7 Feb 2025 23:13:53 +0000 Subject: [PATCH] Enable quantized activation backward kernels in operation-based API tests Signed-off-by: Tim Moon --- tests/pytorch/test_fusible_ops.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/pytorch/test_fusible_ops.py b/tests/pytorch/test_fusible_ops.py index 97d48e2aa3..dbdb93d2db 100644 --- a/tests/pytorch/test_fusible_ops.py +++ b/tests/pytorch/test_fusible_ops.py @@ -1460,6 +1460,7 @@ def test_activation( swiglu=te_ops.SwiGLU, )[activation] forward = te_ops.Sequential( + te_ops.Quantize(forward=False, backward=quantized_compute), make_op(), te_ops.Quantize(forward=quantized_compute, backward=False), )