Add --op-collection option (#2503)
Summary:
This PR add `--op-collection` to tritonbench. It can run multiple ops in defined operator collections. The default collection includes all ops not included in other collections.
Operator collections are defined in `torchbenchmark/operators_collection/`. For each collection, you should define a `get_operators` function to return operators included in this collection.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/2503
Reviewed By: xuzhao9
Differential Revision: D64359380
Pulled By: FindHao
fbshipit-source-id: c66dd254a3c8b70c112d9b7774482813e0236789
