From 2068235c0addeadddb699b62c294dbbd33e24d51 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Mon, 10 Mar 2025 13:40:41 +0000
Subject: [PATCH] Add timm_efficientnet to flaky models after cuda 12.6 update
 in CI/CD (#148788)

After https://github.com/pytorch/pytorch/pull/148612
This model have become flaky

Tracking this regression in an issue : https://github.com/pytorch/pytorch/issues/148699

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148788
Approved by: https://github.com/izaitsevfb, https://github.com/malfet
---
 benchmarks/dynamo/check_accuracy.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/benchmarks/dynamo/check_accuracy.py b/benchmarks/dynamo/check_accuracy.py
index 2de5f86aaa8..7fa24ae7346 100644
--- a/benchmarks/dynamo/check_accuracy.py
+++ b/benchmarks/dynamo/check_accuracy.py
@@ -12,6 +12,7 @@ flaky_models = {
     "yolov3",
     "gluon_inception_v3",
     "detectron2_maskrcnn_r_101_c4",
+    "timm_efficientnet",  # see https://github.com/pytorch/pytorch/issues/148699
     "XGLMForCausalLM",  # discovered in https://github.com/pytorch/pytorch/pull/128148
 }