Skip to content

Commit 447db50

Browse files
committed
fix
1 parent d455ed0 commit 447db50

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

.ci/scripts/validate.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ function generate_aoti_model_output() {
116116

117117
echo "Local DTYPES=$DTYPES"
118118

119-
if [[ DTYPES="default" ]]; then
119+
if [[ DTYPES == "default" ]]; then
120120
if [[ $CHECKPOINT_PATH != *"stories"* && $TARGET_DEVICE == "cuda" ]]; then
121121
DTYPES="bfloat16"
122122
EXCLUDE_INT8_QUANT=true
@@ -126,6 +126,8 @@ function generate_aoti_model_output() {
126126
fi
127127
fi
128128

129+
echo "Local after default DTYPES=$DTYPES"
130+
129131
for DTYPE in $DTYPES; do
130132
echo ""############### Run inference with AOT Inductor for dtype $DTYPE "###############"
131133
echo ""

0 commit comments

Comments
 (0)