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

Install rocm nightly #64

Closed xuzhao9 closed 1 day ago

xuzhao9 commented 1 day ago

We are building rocm nightly docker on-demand, also update the third-party kernel inventories with their dependencies.

Test plan: CI

Fixes https://github.com/pytorch-labs/tritonbench/issues/49

facebook-github-bot commented 1 day ago

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

facebook-github-bot commented 1 day ago

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

facebook-github-bot commented 1 day ago

@xuzhao9 merged this pull request in pytorch-labs/tritonbench@d5702ef86d1a7b5771734dc1328174638a7a406b.