flash-algo
diff --git a/‎benchmarks/benchmark_forward_equivalence.py‎
Lines changed: 1 addition & 1 deletion b/‎benchmarks/benchmark_forward_equivalence.py‎
Lines changed: 1 addition & 1 deletion
@@ -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")
Original file line number	Diff line number	Diff line change
`@@ -377,7 +377,7 @@ def test_forward_equivalence(accuracy_threshold=0.95):`
`377`	`377`	`(1, 2, 1, 64, 64, 128, True),`
`378`	`378`	`(1, 2, 1, 128, 128, 128, True),`
`379`	`379`	`(1, 2, 1, 256, 256, 128, True),`
`380`		`- (1, 2, 1, 512, 512, 128, False),`
	`380`	`+ (1, 2, 1, 512, 512, 128, True),`
`381`	`381`	`]`
`382`	`382`
`383`	`383`	`device = torch.device("cuda" if torch.cuda.is_available() else "cpu")`