How to configure Pytorch to run with NNPACK?

Hi,

I’m trying to build pytorch from source and want to run it with NNPACK.

I tried to export USE_NNPACK=1 before python setup.py install, but this does not take effect (I don’t see it has compiled NNPACK and I cannot find the built nnpack library).

May I know how do I achieve this?