Блог

Главная - Вопросы по программированию - Проверка наборов команд, с помощью которых скомпилирована сборка tensorflow

Проверка наборов команд, с помощью которых скомпилирована сборка tensorflow

#python #tensorflow #build #cpu #instruction-set

#python #tensorflow #сборка #процессор #набор инструкций

Вопрос:

Я ищу способ подтвердить, что колесо tensorflow было скомпилировано с правильными инструкциями.

Я пытаюсь собрать tensorflow для процессора Intel Atom. Я знаю, что bazel автоматически обнаружит и соберет с соответствующими инструкциями, но я не хочу запускать сборку на процессоре Atom, потому что это очень медленно. Итак, я запускаю bazel на процессоре i7 с пользовательскими инструкциями вместо этого. После получения колеса я хотел бы подтвердить, что инструкции действительно использовались.

Я нашел некоторый python api, который позволяет мне проверять MKL и другие материалы, но не знаю, как проверять инструкции процессора, такие как sse. Любая помощь, пожалуйста?

 from tensorflow.python.framework import test_util as test

if __name__ == "__main__":
    print('MKL enabled: %s' % test.IsMklEnabled())
    print('Google cuda enabled: %s' % test.IsGoogleCudaEnabled())
    print('GPU available: %s' % test.is_gpu_available())

    # cpu instruction set?