 Tri Dao
|
bedf877467
[CrossEntropy] Fix where labels address not aligned to 16 bytes
|
2 月之前 |
|
Tri Dao
|
8c20cfef49
[Rotary] Support qkv block layout from GQA
|
3 月之前 |
|
Tri Dao
|
c7f32a8409
[CrossEntropy] Support precomputed LSE
|
3 月之前 |
|
Tri Dao
|
d79f9b41a8
[CrossEntropy] Use online softmax to simplify implementation
|
3 月之前 |
 lancerts
|
22339db185
remove an unused import (#960)
|
6 月之前 |
|
Tri Dao
|
ec6d22143b
[CrossEntropy] Change ignored_index -> ignore_index
|
7 月之前 |
 Curtis "Fjord" Hawthorne
|
d8aacc510c
return z_loss (#768)
|
10 月之前 |
|
Tri Dao
|
08124c8f9c
[CrossEntropy] Implement logit_scale option
|
1 年之前 |
|
Tri Dao
|
aaa1474129
[CrossEntropy] Simplify the case of large vocab with Tensor Parallel
|
1 年之前 |
 Shijie
|
abf04a56e1
fix flash ce mp large vocab (#673)
|
1 年之前 |
|
Tri Dao
|
c79de85ffa
[CrossEntropy] Fix triton cross_entropy_loss IMA for >=2B elements
|
1 年之前 |
|
Tri Dao
|
5400fdc4ac
[CE] Implement CrossEntropyLoss in Triton
|
1 年之前 |
