File test_mea.py

SwissArmyTransformer/tests/test_mea.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	from sat.ops import memory_efficient_attention
2	import torch
3
4	def test_mea(seq_len, hidden_size, num_heads, batch_size=8, qkv=None):

nothing calls this directly

test_mixinFunction · 0.85

no test coverage detected