diff --git a/benchmarks/dynamo/common.py b/benchmarks/dynamo/common.py index 057be601a18..c7eaa77c501 100644 --- a/benchmarks/dynamo/common.py +++ b/benchmarks/dynamo/common.py @@ -252,6 +252,7 @@ CI_SKIP[CI("inductor", training=True, dynamic=True)] = [ *CI_SKIP[CI("inductor", training=False, dynamic=True)], *CI_SKIP[CI("inductor", training=True)], "yolov3", # Accuracy failed torch.Size([4, 3, 12, 16, 85]) + "levit_128", # Accuracy fails on A10G, passes on A100 ] CI_SKIP_OPTIMIZER = {