From f31a083912f3d28b0d9b7419e355dde1df48e8be Mon Sep 17 00:00:00 2001 From: David Corvoysier Date: Thu, 24 Oct 2024 16:06:20 +0000 Subject: [PATCH] test(compile): subclass fix did not make it to 2.5 --- test/tensor/activations/test_activations_compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/tensor/activations/test_activations_compile.py b/test/tensor/activations/test_activations_compile.py index 51b986c4..1fe56598 100644 --- a/test/tensor/activations/test_activations_compile.py +++ b/test/tensor/activations/test_activations_compile.py @@ -27,7 +27,7 @@ def compile_for_device(f, device): return torch.compile(f, backend=backend) -@torch_min_version("2.5.0") +@torch_min_version("2.6.0") @pytest.mark.parametrize("input_shape", [(2, 10), (10, 32, 32)]) @pytest.mark.parametrize("qtype", [qint8], ids=["qint8"]) @pytest.mark.parametrize("dtype", [torch.float32, torch.float16, torch.bfloat16], ids=["fp32", "fp16", "bf16"])