issues
search
pytorch-labs
/
tritonbench
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
BSD 3-Clause "New" or "Revised" License
20
stars
3
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Add pr test to run operators on h100
#25
xuzhao9
closed
3 weeks ago
2
Deploy the H100 runner infra
#24
xuzhao9
closed
3 weeks ago
3
Add xformers submodule
#23
xuzhao9
closed
3 weeks ago
4
Fix loading ThunderKitten libraries
#22
xuzhao9
closed
3 weeks ago
2
Add custom ops fused_linear_cross_entropy,geglu,cross_entropy
#21
FindHao
closed
3 weeks ago
2
Add xformers support
#20
xuzhao9
closed
1 day ago
12
Remove hammer/generative_recommenders
#19
xing-liu
closed
3 weeks ago
5
Re-sync with internal repository
#18
facebook-github-bot
closed
4 weeks ago
0
[Installation][non-reproducible]: Op Flash Attention
#17
antferdom
closed
2 days ago
15
Add custom ops jsd and fused_linear_jsd
#16
FindHao
closed
3 weeks ago
1
Add embedding,rms_norm,rope
#15
FindHao
closed
3 weeks ago
4
Add custom ops kl_div, swiglu
#14
FindHao
closed
3 weeks ago
3
Add custom ops fused_linear_cross_entropy,geglu,cross_entropy
#13
FindHao
closed
3 weeks ago
3
Add back submodules and ci
#12
xuzhao9
closed
3 weeks ago
1
Re-sync with internal repository
#11
facebook-github-bot
closed
1 month ago
0
Check in benchmarks directory
#10
xuzhao9
closed
1 month ago
0
Add operator collection and operator loaders
#9
xuzhao9
closed
1 month ago
0
Add operators from pytorch/benchmark
#8
xuzhao9
closed
1 month ago
0
Compile and install triton main branch in the docker.
#7
xuzhao9
closed
1 month ago
0
Use 32-core-ubuntu runner
#6
xuzhao9
closed
1 month ago
0
Skip fa2 compile in docker
#5
xuzhao9
closed
1 month ago
0
Fix docker install flash-attn
#4
xuzhao9
closed
1 month ago
0
Fix the docker build
#3
xuzhao9
closed
1 month ago
0
Add installation scripts.
#2
xuzhao9
closed
1 month ago
0
Add submodules
#1
xuzhao9
closed
1 month ago
0
Previous