Skip to content

Commit

Permalink
change matmul size
Browse files Browse the repository at this point in the history
  • Loading branch information
rdspring1 committed Aug 31, 2024
1 parent aa9e9cd commit 795255d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/python/opinfo_input_generators.py
Original file line number Diff line number Diff line change
Expand Up @@ -1492,9 +1492,9 @@ def matmul_input_generator(
requires_grad=requires_grad,
)

B = 64
M = 512
N = 256
B = 8
M = 256
N = 128
K = 32

shapes_a = ((K,), (M, K), (1, K), (B, M, K), (B, 1, M, K))
Expand Down

0 comments on commit 795255d

Please sign in to comment.