Skip to content

Commit

Permalink
fix spacing again
Browse files Browse the repository at this point in the history
  • Loading branch information
sashaDoubov committed Nov 30, 2023
1 parent c5832ef commit ee5ff20
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion megablocks/layers/glu_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,6 @@ class GLUTest(parameterized.TestCase):

@parameterized.parameters(*_DENSE_TESTS)
def testGLU_ForwardGroupedMLP(self, bs, sl, hs):

x = torch.randn(sl, bs, hs).to(torch.bfloat16).cuda()

_, glu, dmoe_glu = test_modules(
Expand Down

0 comments on commit ee5ff20

Please sign in to comment.