我正在寻找一种方法来确认是否使用正确的指令编译了张量流轮。
我正在尝试为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?