search

hidet-org / hidet

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

[Library] Add cublas library #404

Closed yaoyaoding closed 9 months ago

yaoyaoding commented 9 months ago

This PR add cublas to hidet.

Check the tests/cuda/test_cublas.py for the usage of cublas in hidet.