[Dev] Feature: linear cross entropy fusion - #2256

Merged

shjwudp merged 23 commits into

Jianbing-D:feat-linear-cross-entropy-for-dev

Dec 5, 2025

Commits on Nov 14, 2025

add fused_linear_cross_entropy interface
Jianbing-D
committed
36755be
Merge pull request #1 from shjwudp/jianbinc/fused_linear_ce

shjwudp
authored and
Jianbing-D
committed
5781d3d
Feat linear cross entropy kernel dev (#2)
Jianbing-D
committed
289847f
remove tl.pointer_type() type annotation for triton kernels (#3)
Jianbing-D
committed
14e5360
Support Tensor Parallel and Sequence Parallel (#4)
Jianbing-D
committed
45dfe42
Fixed several bugs and added support for sequence-parallel (#5)

shjwudp
authored and
Jianbing-D
committed
0d12e00
formatting and fixing lints (#6)
Jianbing-D
committed
24a5465
Remove redundant logits calculations in gpt_model (#9)

shjwudp
authored and
Jianbing-D
committed
f653838
fixed some styling issue (#10)
Jianbing-D
committed
66d43ff
rename compute_output_layer_and_language_model_loss (#12)

shjwudp
authored and
Jianbing-D
committed
c1548f8

Commits on Nov 17, 2025

Commits on Nov 21, 2025

remove unrelated change (#13)
shjwudp
authored
5003264
Merge branch 'dev' into feat-linear-cross-entropy-for-dev
shjwudp
authored
65a98b7
Merge branch 'dev' into feat-linear-cross-entropy-for-dev
Jianbing-D
authored
4035083

Commits on Nov 27, 2025

handle non-blackwell arch platform init fail (#14)
shjwudp
authored
78c827e
Remove the migration code from the main branch to the dev branch (#15)
shjwudp
authored
011947d
Fix review (#16)
Jianbing-D
authored
bfee2a3

Commits on Dec 2, 2025

Fix for CI (#17)
shjwudp
authored
1b603b9
Fix CI (#18)
shjwudp
authored
fb2ee78
Update Dev Branch & Fix CI (#19)

authored
48c5228
Merge branch 'dev' into feat-linear-cross-entropy-for-dev
shjwudp
authored
4dc347c

Commits on Dec 3, 2025

Merge branch 'dev' into feat-linear-cross-entropy-for-dev
shjwudp
authored
ae4d83a