Скомпилируйте пакет CPX mxnet с mkldnn для поддержки однопотокового выполнения без переменных среды - PullRequest
2 голосов
/ 06 ноября 2019

Мы ищем способ использовать mxnet CPP lib с одним потоком в Linux-подобных системах. Из документации mxnet и опыта сообщества мы нашли единственный способ установить переменные среды:

export MXNET_ENGINE_TYPE=NaiveEngine
export OMP_NUM_THREADS=1

Но этот способ проблематичен, не всегда возможно настроить переменные среды.

Углубление вкод mxnet, который мы обнаружили, что есть некоторые определения препроцессора и флаги CMake, которые могут помочь

mxnet  preprocessor definition MXNET_PREDICT_ONLY=1 will enable naiveEngine
mkldnn cmake flag MKLDNN_THREADING=SEQ

Вопросы:

1) Возможно ли вообще скомпилировать mxnet в однопоточном режиме?

2) Если это возможно, какое минимальное количество изменений мы можем сделать, чтобы скомпилировать однопоточную библиотеку mxnet?

3) Есть ли какое-либо другое решение, обеспечивающее однопоточное выполнение mxnetмодель нейронной сети?

Любая помощь будет оценена.

...