search

warmspringwinds / pytorch-cpp

Pytorch C++ Library
369 stars 68 forks source link

how to turn on intel MKL for BLAS? #2

Open wangzhenhua2015 opened 6 years ago

wangzhenhua2015 commented 6 years ago

In CPU mode, what is the BLAS as default? how to support intel MKL?

Thanks, Adam Wang

warmspringwinds commented 6 years ago

Hi @wangzhenhua2015

We mainly used this for GPU only so far.

But I am pretty sure this can be done when compiling the ATen library (which is the main dependency).

I suggest you open an issue asking this question: https://github.com/zdevito/ATen