23 Commits

Author	SHA1	Message	Date
Joshua Ferguson	4a70a0f8bc	renaming FloatTensor Ops, Primitives, and maybe functions (#1174)	2024-01-27 10:04:50 -05:00
Joshua Ferguson	3b7d9feede	Elementwise pow op (#1133)	2024-01-24 09:46:57 -05:00
Nathaniel Simard	635cc0a3e7	[burn-fusion] overlapping optimization builders (#1162)	2024-01-22 16:18:59 -05:00
Nathaniel Simard	ebb2306a6b	[burn-fusion] support multiple concurrent optimization streams (#1149)	2024-01-18 13:15:43 -05:00
Nathaniel Simard	b99726f804	[burn-fusion] save all execution plans for any trigger (#1143)	2024-01-16 14:02:42 -05:00
Nathaniel Simard	0a85744d9b	Refactor/fusion stream (#1135)	2024-01-13 14:17:43 -05:00
Nathaniel Simard	f0e10f1499	Refactor/fusion optimization (#1104)	2024-01-06 14:22:50 -05:00
unrenormalizable	40ec289a92	Making Embedding.weight public (#1094)	2023-12-22 18:11:38 -05:00
Nathaniel Simard	b0a2b30ed1	Fix/wgpu/fusion (#1093) ... * Fix fusion * Real fix * Fix	2023-12-22 16:53:19 -05:00
Nathaniel Simard	b5c49c5bf7	Fusion wgpu compilation cache (#1069) ... * Refactor fusion in the wgpu backend * WIP * Refactor * WIP * Fix inplace ops * Works ish * Cleanup Output * Refactoring * Refactor Clamp * Cleanup * Cleanup * Updates * Fix CI * Code review	2023-12-18 12:16:08 -05:00
David Chavez	71d3c1d142	chore(infra): Share some properties across workspace (#1039)	2023-12-12 09:39:07 -05:00
Louis Fortier-Dubois	8fc52113bc	Chore/bump v12 (#1048)	2023-12-04 10:47:54 -05:00
Louis Fortier-Dubois	3088c466a5	patch 0.11.1 (#1047)	2023-12-04 10:18:30 -05:00
Nathaniel Simard	0b16786729	Fix/orphan cleanup (#1043)	2023-12-04 09:44:17 -05:00
Timothy	52811f9938	Fix some issues in Burn book (#1042)	2023-12-04 09:23:47 -05:00
Nathaniel Simard	ab1b5890f5	Chore/release (#1031)	2023-12-01 14:33:28 -05:00
Nathaniel Simard	670280dda2	Feat/fusion/cache (#1020)	2023-12-01 12:05:11 -05:00
Nathaniel Simard	f6d14f1b1a	Refactor feature flags (#1025)	2023-12-01 09:48:28 -05:00
Nathaniel Simard	3d6c738776	Refactor/fusion/graph (#988)	2023-11-22 09:55:42 -05:00
Nathaniel Simard	24014aca33	WGPU: Support elemwise operation fusion (#948)	2023-11-15 15:13:37 -05:00
Zsombor	4fc0c27e31	Implement tensor.recip() function to calculate elementwise reciprocals (#953)	2023-11-15 09:17:32 -05:00
Louis Fortier-Dubois	4d63a24e19	fix clippy (#955)	2023-11-13 15:30:07 -05:00
Nathaniel Simard	322480b744	Feat/op fusion decorator (#939) ... * WIP * Impl backend decorator * WIP * WIP * WIP * WIP * WIP * WIP * Refactor * Handle graph single ops execution * WIP * Starting to get concrete * WIP * Fix locator * Implement add ops * Start implementing ops * Add more ops * Add more ops * More float ops * Almost finish float ops * Almost done with Int * Some fix * Into float * Implement bool ops * Almost done with MVP * Fix adaptive pooling * Add fusion as backend * Fix memory leak * Fix * WIP Doc * Doc all ops enum * Initial docs * Clippy * Clippy v2 * Fix typos * Fix doc * Fix feature flags * Add missing ops * Some cleanup * Revert u128 id * cosmetic fixes --------- Co-authored-by: louisfd <louisfd94@gmail.com>	2023-11-09 21:21:41 -05:00