Проверьте наборы команд Сборка тензорного потока компилируется с - PullRequest
Новая
       46

Проверьте наборы команд Сборка тензорного потока компилируется с

1 голос
/ 25 марта 2019

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

Я пытаюсь построить тензор потока для процессора 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?
...