search

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

[Fixbug] Fix for softmmax cpu causing issues #437

Open fishingguy456 opened 4 months ago

fishingguy456 commented 4 months ago

moved implement_cpu to the cpu task