From 4ae486530f1f77be5321e073f9f0b8eac56bcf24 Mon Sep 17 00:00:00 2001 From: Loser Cheems Date: Tue, 1 Jul 2025 12:40:29 +0800 Subject: [PATCH] Enables test case for 512x512 input dimensions Changes the boolean flag from False to True for the largest input size test case, allowing the forward equivalence benchmark to include 512x512 resolution testing in its validation suite. --- benchmarks/benchmark_forward_equivalence.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/benchmark_forward_equivalence.py b/benchmarks/benchmark_forward_equivalence.py index 2c73a53..5d89419 100644 --- a/benchmarks/benchmark_forward_equivalence.py +++ b/benchmarks/benchmark_forward_equivalence.py @@ -377,7 +377,7 @@ def test_forward_equivalence(accuracy_threshold=0.95): (1, 2, 1, 64, 64, 128, True), (1, 2, 1, 128, 128, 128, True), (1, 2, 1, 256, 256, 128, True), - (1, 2, 1, 512, 512, 128, False), + (1, 2, 1, 512, 512, 128, True), ] device = torch.device("cuda" if torch.cuda.is_available() else "cpu")