1153 Commits

Author	SHA1	Message	Date
Tianxing Wu	98f15eedcf	Added block q	2025-11-05 11:44:10 +00:00
Tianxing Wu	d5c8315aff	fixed window creation number<>{}	2025-10-28 11:32:48 +00:00
Tianxing Wu	22c5c20977	Debugging window size	2025-10-24 09:45:32 +00:00
Tianxing Wu	89cfdb35e0	Fixed block Q with M	2025-10-23 12:02:18 +00:00
Tianxing Wu	d18f8e46bf	Fixed block Q with M	2025-10-23 12:02:10 +00:00
Tianxing Wu	ebf1c4c305	const blockq	2025-10-23 11:57:36 +00:00
Tianxing Wu	3bb29bfd6c	Fixed pipeline args	2025-10-23 11:49:35 +00:00
Tianxing Wu	6ea56bec34	removed redundent code	2025-10-23 11:44:14 +00:00
Juuso Korhonen	5bf72d2bcb	fixing bugs	2025-10-23 11:40:48 +00:00
Tianxing Wu	3bcef59536	block table stride fix	2025-10-23 11:25:07 +00:00
Tianxing Wu	0d2a9badba	fixed example	2025-10-23 11:17:46 +00:00
Juuso Korhonen	3c0e6d37bf	fixing bugs	2025-10-23 09:47:30 +00:00
Juuso Korhonen	e144872308	change to BLOCK_M in shape definitions	2025-10-23 08:11:55 +00:00
Tianxing Wu	f72b994b00	More compilation fixes	2025-10-20 15:53:35 +00:00
Juuso Korhonen	d68a541c19	fixing compile errors...	2025-10-20 15:04:47 +00:00
Juuso Korhonen	97e7527eb1	fixing compile errors...	2025-10-20 14:03:15 +00:00
Tianxing Wu	9fda954253	Compiling fix	2025-10-20 13:16:19 +00:00
Tianxing Wu	995c6701d3	Merge branch 'tianxing/unified-attention' of https://github.com/ROCm/composable_kernel into tianxing/unified-attention	2025-10-17 09:05:12 +00:00
Tianxing Wu	af9167abad	example	2025-10-17 09:05:10 +00:00
Juuso Korhonen	9940bd07f6	fix order in mask caller	2025-10-16 11:23:46 +00:00
Juuso Korhonen	072de3842f	comment	2025-10-16 09:23:39 +00:00
Juuso Korhonen	aa4908ac14	fix mask	2025-10-16 09:18:38 +00:00
Juuso Korhonen	62932576c4	use correct mask in kernel	2025-10-16 09:02:08 +00:00
Juuso Korhonen	498a97aa1d	merge	2025-10-16 08:57:14 +00:00
Juuso Korhonen	63c17b7236	correct masking by transforming y_idx = y_idx / num_queries_per_kv	2025-10-16 08:54:07 +00:00
Tianxing Wu	853fa21566	Example boostrap	2025-10-15 11:58:44 +00:00
Juuso Korhonen	72fe8b311c	merge	2025-10-14 12:35:33 +00:00
Juuso Korhonen	4d232d59cc	fix seq_len -> cur_batch_query_len	2025-10-14 12:34:33 +00:00
Tianxing Wu	b940a75328	Comments	2025-10-14 12:19:20 +00:00
Tianxing Wu	ec29289bb1	kv paging	2025-10-14 12:04:11 +00:00
Tianxing Wu	c87f2e3ca9	o window change	2025-10-14 09:59:47 +00:00
Tianxing Wu	96b208f6c7	Merge branch 'tianxing/unified-attention' of https://github.com/ROCm/composable_kernel into tianxing/unified-attention	2025-10-14 09:58:30 +00:00
Tianxing Wu	e1120fffb0	pipeline api	2025-10-14 09:58:27 +00:00
Juuso Korhonen	c3d27abfb8	fix q window	2025-10-14 09:49:54 +00:00
Juuso Korhonen	b37c356090	fix q window origin	2025-10-14 09:36:28 +00:00
Tianxing Wu	6a7fa959b7	kv tensor view and initial window	2025-10-13 12:53:43 +00:00
Tianxing Wu	cd354286c1	Merge branch 'tianxing/unified-attention' of https://github.com/ROCm/composable_kernel into tianxing/unified-attention	2025-10-13 11:32:30 +00:00
Tianxing Wu	be58d51d36	o ptr and window	2025-10-13 11:32:28 +00:00
Juuso Korhonen	6ba25b7e84	add commenting	2025-10-13 10:34:55 +00:00
Juuso Korhonen	81a02ffb40	Merge branch 'tianxing/unified-attention' of https://github.com/ROCm/composable_kernel into tianxing/unified-attention	2025-10-13 10:30:22 +00:00
Juuso Korhonen	b721f79f99	fix	2025-10-13 10:30:11 +00:00
Tianxing Wu	16129a794a	stride fix	2025-10-13 10:30:08 +00:00
Tianxing Wu	96fde33ec4	Merge branch 'tianxing/unified-attention' of https://github.com/ROCm/composable_kernel into tianxing/unified-attention	2025-10-13 10:29:07 +00:00
Tianxing Wu	55fc6d7151	kv tensor view	2025-10-13 10:28:02 +00:00
Juuso Korhonen	af94aaf1cb	refactor the q tensor view transformation	2025-10-13 10:22:52 +00:00
Juuso Korhonen	49ce980c67	Merge branch 'tianxing/unified-attention' of https://github.com/ROCm/composable_kernel into tianxing/unified-attention	2025-10-13 10:21:27 +00:00
Juuso Korhonen	2d6dab29eb	refactor the q tensor view transformation	2025-10-13 10:18:23 +00:00
Tianxing Wu	36a65b1968	refactor	2025-10-13 10:05:23 +00:00
Tianxing Wu	bc6385f389	Some refactor	2025-10-13 10:01:38 +00:00
Tianxing Wu	1f4648dab5	refactor. and fixed q transformation	2025-10-10 15:27:36 +00:00