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
Remaining works for liger kernels
#31
Closed
FindHao
closed
3 weeks ago
FindHao
commented
3 weeks ago
[x] Add layernorm kernel
[ ] Change all ops' x_val to input shapes
[ ] Add memory footprint metric
[ ] format all benchmark names