Ops bug fix and args clean

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

Closed FindHao closed 1 day ago

FindHao commented 1 day ago

Fix rope's bug for specific nsys profile. Clean torch.compile args.

facebook-github-bot commented 1 day ago

@FindHao 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

@FindHao 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

@FindHao merged this pull request in pytorch-labs/tritonbench@ebdb921a6d4fdb343f3e3eebc1de3f35acb6aa1c.