检查指令集tensorflow构建是用

时间:2019-03-25 04:14:14

标签: python tensorflow build cpu instruction-set

我正在寻找一种方法来确认是否使用正确的指令编译了张量流轮。

我正在尝试为Intel Atom CPU构建张量流。我知道bazel会自动检测并使用适当的指令进行构建,但是我不想在Atom CPU上运行该构建,因为它非常慢。因此,我使用自定义指令在i7 CPU上运行bazel。拿到轮子后,我想确认一下说明是否确实在使用。

我找到了一些python API,可以检查MKL和其他内容,但不知道如何检查sse等cpu指令。有什么帮助吗?

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?

0 个答案:

没有答案