search

thomasbrandon / mish-cuda

Mish Activation Function for PyTorch
MIT License
147 stars 67 forks source link

Compile on CPU #16

Open CuongNN218 opened 3 years ago

CuongNN218 commented 3 years ago

can we compile it only for CPU

knotgrass commented 2 years ago

in my view, if you want to runing on CPU, you can using repo.
Because, this repo has written by cuda, so compile on CPU was impossible.