Skip to content

Commit d21baff

Browse files
Jonathan Hseujhseu
Jonathan Hseu
authored andcommitted
Add MAVX2_FMA_DBG build option and rename MAVXDBG to MAVX_DBG.
1 parent 52389ea commit d21baff

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tensorflow/tools/ci_build/ci_parameterized_build.sh

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,8 @@
8484
# support for Google Cloud Platform (GCP), which is
8585
# enabled by default.
8686
# TF_BUILD_OPTIONS:
87-
# (FASTBUILD | OPT | OPTDBG | MAVX | MAVX2)
87+
# (FASTBUILD | OPT | OPTDBG | MAVX | MAVX2 | MAVX_DBG |
88+
# MAVX2_FMA_DBG)
8889
# Use the specified configurations when building.
8990
# When set, overrides TF_BUILD_IS_OPT and TF_BUILD_MAVX
9091
# options, as this will replace the two.
@@ -305,12 +306,15 @@ else
305306
MAVX)
306307
OPT_FLAG="${OPT_FLAG} -c opt --copt=-mavx"
307308
;;
308-
MAVXDBG)
309+
MAVX_DBG)
309310
OPT_FLAG="${OPT_FLAG} -c opt --copt=-g --copt=-mavx"
310311
;;
311312
MAVX2)
312313
OPT_FLAG="${OPT_FLAG} -c opt --copt=-mavx2"
313314
;;
315+
MAVX2_FMA_DBG)
316+
OPT_FLAG="${OPT_FLAG} -c opt --copt=-g --copt=-mavx2 --copt=-mfma"
317+
;;
314318
esac
315319
fi
316320

0 commit comments

Comments
 (0)