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
21 stars 3 forks source link

Add autotune mode to liger kernels #43

Open FindHao opened 2 weeks ago

FindHao commented 4 days ago

there are a number of torch.compile configs we can set, it's better to use the way like what we have done in operator_loader