File tree Expand file tree Collapse file tree 1 file changed +22
-1
lines changed Expand file tree Collapse file tree 1 file changed +22
-1
lines changed Original file line number Diff line number Diff line change @@ -52,6 +52,26 @@ def is_raspi_arm():
5252 return os .uname ()[4 ] == "armv7l"
5353
5454
55+ def is_linux_ppc64le ():
56+ return is_linux () and platform .machine () == "ppc64le"
57+
58+
59+ def is_linux_x86_64 ():
60+ return is_linux () and platform .machine () == "x86_64"
61+
62+
63+ def is_linux_arm ():
64+ return is_linux () and platform .machine () == "arm"
65+
66+
67+ def is_linux_aarch64 ():
68+ return is_linux () and platform .machine () == "aarch64"
69+
70+
71+ def is_linux_s390x ():
72+ return is_linux () and platform .machine () == "s390x"
73+
74+
5575def get_tf_header_dir ():
5676 import tensorflow as tf
5777
@@ -121,7 +141,8 @@ def create_build_configuration():
121141 write ("build:windows --host_cxxopt=/std:c++14" )
122142
123143 if is_macos () or is_linux ():
124- write ("build --copt=-mavx" )
144+ if not is_linux_ppc64le ():
145+ write ("build --copt=-mavx" )
125146 write ("build --cxxopt=-std=c++14" )
126147 write ("build --host_cxxopt=-std=c++14" )
127148
You can’t perform that action at this time.
0 commit comments