4 Commits

Author	SHA1	Message	Date
Liangfu Chen	d2b58ca203	[Neuron][kernel] Fuse kv cache into a single tensor (#15911) ... Signed-off-by: Liangfu Chen <liangfc@amazon.com>	2025-04-03 09:51:32 -07:00
Liangfu Chen	53a0cf8b95	[Neuron] trim attention kernel tests to fit trn1.2x instance (#14988) ... Signed-off-by: Liangfu Chen <liangfc@amazon.com>	2025-03-18 15:05:52 +08:00
Sibi	a73e183e36	[Misc] Replace os environ to monkeypatch in test suite (#14516) ... Signed-off-by: sibi <85477603+t-sibiraj@users.noreply.github.com> Signed-off-by: Aaron Pham <contact@aarnphm.xyz> Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com> Co-authored-by: Aaron Pham <contact@aarnphm.xyz>	2025-03-16 20:35:57 -07:00
Liangfu Chen	d3d4956261	[Neuron] flatten test parameterization for neuron attention kernels (#14712)	2025-03-13 20:46:56 -07:00