hidet-org / hidet

An open-source efficient deep learning framework/compiler, written in python.
https://hidet.org
Apache License 2.0
634 stars 50 forks source link

Google Colab: KeyError Primitive function cuda_i64_to_f16 has already registered #444

Open egorsmkv opened 3 months ago

egorsmkv commented 3 months ago

Describe the bug

I am trying to install hidet 0.3.1 on Google Colab

To Reproduce

pip install https://github.com/hidet-org/hidet/archive/refs/tags/v0.3.1.zip
import hidet

print('hidet:', hidet.__version__)

Expected behavior

To see the hidet version

Additional context

Instead of the hidet version I am getting this error:

image