Tensorflow 与 bazel：avx 被忽略？

Question

我正在尝试手动编译张量流以利用 avx。不幸的是，我似乎专门忽略了这些说明。我可能会补充一点，我正在 Windows 10 上工作。任何帮助将不胜感激。谢谢。

我对 bazel 使用了以下命令：

bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both -k //tensorflow/tools/pip_package:build_pip_package

Answer 1

-mavx等等仅适用于 GCC 和 Clang 编译器。由于您使用的是 MSVC，因此您应该使用--copt=/arch:AVX --copt=/arch:AVX2等等。