Skip to content

Commit d39682b

Browse files
committed
Increase tiemout for gelu
1 parent 8d31e25 commit d39682b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/nn.spec.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -684,7 +684,8 @@ describe('layers', function() {
684684
assertArrayAllTrue(mx.arrayEqual(y, y2));
685685
});
686686

687-
it('gelu', () => {
687+
it('gelu', function() {
688+
this.timeout(20 * 1000);
688689
const inputs = mx.array([1.15286231, -0.81037411, 0.35816911, 0.77484438, 0.66276414]);
689690
const expected = mx.array([1.0093501, -0.16925684, 0.22918941, 0.60498625, 0.49459383]);
690691
const expectedApprox = mx.array([1.0091482, -0.1693441, 0.22918446, 0.60491, 0.4945476]);

0 commit comments

Comments
 (0)