Function
naive_method
(vocab_size: int, hidden_size: int, n_hidden_states: int)
Source from the content-addressed store, hash-verified
| 16 | |
| 17 | |
| 18 | def naive_method(vocab_size: int, hidden_size: int, n_hidden_states: int): |
| 19 | n = n_hidden_states |
| 20 | m = vocab_size |
| 21 | k = hidden_size |
| 22 | # first stage: matmul |
| 23 | reads1 = m * k + n * k |
| 24 | writes1 = m * n |
| 25 | # second stage: sampling |
| 26 | reads2 = writes1 |
| 27 | writes2 = n |
| 28 | return reads1 + reads2, writes1 + writes2 |
| 29 | |
| 30 | |
| 31 | class Args(BaseSettings): |
Tested by
no test coverage detected