PyTorch Build - AshokBhat/ml GitHub Wiki
PyTorch Build flags
| Flag | Description |
|---|---|
| USE_MKL=ON | - |
| USE_MKLDNN=ON | |
| USE_NNPACK=ON | |
| USE_OPENMP=ON | |
| USE_CUDA=ON | |
| PERF_WITH_AVX512=1 | |
| BLAS=MKL | |
| BUILD_TYPE=Release |
Build defines
| Flag | Description |
|---|---|
| -DUSE_PTHREADPOOL | |
| -fopenmp | |
| -DUSE_FBGEMM | |
| -DUSE_QNNPACK | |
| -DUSE_PYTORCH_QNNPACK | |
| -DUSE_XNNPACK |