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 xformers submodule #23

Closed xuzhao9 closed 4 weeks ago

xuzhao9 commented 4 weeks ago

Test plan:

$ python install.py --xformers
$ python -c "import xformers.ops; op = xformers.ops.fmha.flash3.FwOp; print(op.is_available())"
True
xuzhao9 commented 4 weeks ago

cc @antferdom

facebook-github-bot commented 4 weeks ago

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

facebook-github-bot commented 4 weeks ago

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

facebook-github-bot commented 4 weeks ago

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