Skip to content

Commit

Permalink
[test] scale up accuracy test shapes
Browse files Browse the repository at this point in the history
  • Loading branch information
StrongSpoon committed May 31, 2024
1 parent a681112 commit 0b964e2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/accuracy_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
}

POINTWISE_SHAPES = [(1024, 1024), (16, 1024, 256), (16, 128, 64, 64), (20, 320, 15)]
REDUCTION_SHAPES = [(1024, 64 * i) for i in range(1, 10, 2)]
REDUCTION_SHAPES = [(4096, 256 * i) for i in range(1, 10, 2)]
MNK_SHAPES = [15, 160, 1024]

FLOAT_DTYPES = [torch.float16, torch.float32, torch.bfloat16]
Expand Down

0 comments on commit 0b964e2

Please sign in to comment.