6 Commits

Author	SHA1	Message	Date
Mor Zusman	fb60ae9b91	[Kernel][Model] Improve continuous batching for Jamba and Mamba (#9189)	2024-10-16 12:12:43 -04:00
Mor Zusman	f13a07b1f8	[Kernel][Model] Varlen prefill + Prefill chunking support for mamba kernels and Jamba model (#8533)	2024-09-29 17:35:58 -04:00
bnellnm	300da09177	[Kernel] Fullgraph and opcheck tests (#8479)	2024-09-25 08:35:52 -06:00
Tyler Michael Smith	db9120cded	[Kernel] Change interface to Mamba selective_state_update for continuous batching (#8039)	2024-09-18 20:05:06 +00:00
Cyrus Leung	6ffa3f314c	[CI/Build] Avoid CUDA initialization (#8534)	2024-09-18 10:38:11 +00:00
Mor Zusman	fdd9daafa3	[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651)	2024-08-28 15:06:52 -07:00