Add an init file to tools so that it is considered as an module.

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

Add an init file to tools so that it is considered as an module. #48

Closed htyu closed 1 week ago

htyu commented 1 week ago

Fixing issues like:

File "tritonbench/install.py", line 8, in from tools.cuda_utils import CUDA_VERSION_MAP, DEFAULT_CUDA_VERSION ModuleNotFoundError: No module named 'tools.cuda_utils'

facebook-github-bot commented 1 week ago

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

facebook-github-bot commented 1 week ago

@htyu merged this pull request in pytorch-labs/tritonbench@d2f37a421ca76581d44a0ad10f517e24277ba0fe.