Description
Describe the bug
I'm trying to optimize pytorch models with hidet. However, whether installing hidet via pip or from source, it will produce an error that the backend cannot be found:
File "cell1.py", line 12, in
model_opt = torch.compile(model, backend='hidet')
File "/usr/local/lib/python3.8/dist-packages/torch/init.py", line 1419, in compile
return torch._dynamo.optimize(backend=backend, nopython=fullgraph, dynamic=dynamic, disable=disable)(model)
File "/usr/local/lib/python3.8/dist-packages/torch/_dynamo/eval_frame.py", line 426, in optimize
backend = get_compiler_fn(backend)
File "/usr/local/lib/python3.8/dist-packages/torch/_dynamo/eval_frame.py", line 356, in get_compiler_fn
compiler_fn = lookup_backend(compiler_fn)
File "/usr/local/lib/python3.8/dist-packages/torch/_dynamo/eval_frame.py", line 365, in lookup_backend
compiler_fn = BACKENDS[compiler_fn]
KeyError: 'hidet'
To Reproduce
gallery/getting-started/quick-start.py
Expected behavior
Theoretically, torch dynamo should have registered the hidet backend after successful installation. Do you have any idea about it?
Enviroment
- Python verison: 3.8.10
- Hidet version: 0.2.2.dev
- PyTorch version: 2.0.0+cu116
- OS: Ubuntu 22.04.2 LTS
- GPU: NVIDIA A10
- GPU driver: 525.85.12