7 Commits

Author	SHA1	Message	Date
HELSON	537e181705	[testing] fix testing models (#2036) ... * [testing] fix testing models * roll back	2022-11-29 13:42:06 +08:00
Ziyue Jiang	75d221918a	[Tensor] add 1d vocab loss (#918) ... * add 1d vocab loss * polish	2022-05-07 15:49:14 +08:00
Ziyue Jiang	f593a5637e	[Tensor] add embedding tp1d row (#904)	2022-04-29 14:10:05 +08:00
Jiarui Fang	1190b2c4a4	[tensor] add cross_entrophy_loss (#868)	2022-04-25 16:01:52 +08:00
Frank Lee	1258af71cc	[ci] cache cuda extension (#860)	2022-04-25 10:03:47 +08:00
Ziyue Jiang	bcc8655021	[Tensor ] Add 1Drow weight reshard by spec (#854)	2022-04-24 18:30:20 +08:00
Jiarui Fang	62f059251b	[Tensor] init a tp network training unittest (#849)	2022-04-24 16:43:44 +08:00