14 Commits

Author	SHA1	Message	Date
Nikola Borisov	87069ccf68	Fix docker python version (#2845)	2024-02-14 10:17:57 -08:00
Simon Mo	b9e96b17de	fix python 3.8 syntax (#2716)	2024-02-01 14:00:58 -08:00
Philipp Moritz	d0d93b92b1	Add unit test for Mixtral MoE layer (#2677)	2024-01-31 14:34:17 -08:00
Philipp Moritz	390b495ff3	Don't build punica kernels by default (#2605)	2024-01-26 15:19:19 -08:00
Simon Mo	6e01e8c1c8	[CI] Add Buildkite (#2355)	2024-01-14 12:37:58 -08:00
Alexandre Payot	937e7b7d7c	Build docker image with shared objects from "build" step (#2237)	2024-01-04 09:35:18 -08:00
Antoni Baum	21d93c140d	Optimize Mixtral with expert parallelism (#2090)	2023-12-13 23:55:07 -08:00
Simon Mo	3fefe271ec	Update Dockerfile to build Megablocks (#2042)	2023-12-12 17:34:17 -08:00
Simon Mo	eb17212858	Update Dockerfile to support Mixtral (#2027)	2023-12-11 11:59:08 -08:00
Simon Mo	c85b80c2b6	[Docker] Add cuda arch list as build option (#1950)	2023-12-08 09:53:47 -08:00
AguirreNicolas	24f60a54f4	[Docker] Adding number of nvcc_threads during build as envar (#1893)	2023-12-07 11:00:32 -08:00
Allen	f07c1ceaa5	[FIX] Fix docker build error (#1831) (#1832) ... Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>	2023-11-29 23:06:50 -08:00
GhaziSyed	aebfcb262a	Dockerfile: Upgrade Cuda to 12.1 (#1609)	2023-11-09 11:49:02 -08:00
Stephen Krider	9cabcb7645	Add Dockerfile (#1350)	2023-10-31 12:36:47 -07:00