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

Use 32-core-ubuntu runner #6

Closed xuzhao9 closed 1 month ago

xuzhao9 commented 1 month ago

Make the following changes:

  1. Add back the FA2 package and compile from source code
  2. Use 32-core-ubuntu runner to build the docker
  3. Add workflow file to the GHA workflow path

Test plan: GHA CI